形成 n 位数字范围的更优雅的方式

More elegant way to form n-digit range of numbers

我需要根据输入参数 n 形成一个数字范围。如果 n=1 范围是 range(0, 10),如果 n=2 那么 range(10, 100).

我有一个代码,但是看起来很难看,也许有更好的方法。

start = 0 if n == 1 else int('1' + '0' * (n - 1))
end = 10 if n == 1 else 10 * start
for i in range(start, end):

对于n=1应该是:

range(0, 10) 
[0,1,2,3,4,5,6,7,8,9]
>>> range(pow(10, n), pow(10, n)*10)
[1, 2, 3, 4, 5, 6, 7, 8, 9]


>>> range(pow(10, n), pow(10, n)*10)
[10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99]

试试这个:

if n == 1:
  interval = range(0, 10**n)
else:
  interval = range(10**(n-1), 10**n)

如果 n=1 的范围是 range(1,10),您可以只使用:range(10**(n-1), 10**n) 不带 if 子句

这向您展示了如何使用 pow math.pow 来解决您的用例。 它并没有解决每一个细节,但它为您指明了正确的方向。

>>> from math import pow
>>> n = 1
>>> range(int(pow(10, n - 1)), int(pow(10, n)))
[1, 2, 3, 4, 5, 6, 7, 8, 9]
>>> n = 2
>>> range(int(pow(10, n - 1)), int(pow(10, n)))
[10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99]

>>> n = -1
>>> range(int(pow(10, n - 1)), int(pow(10, n)))
[]

您需要确保 n 大于 den 0 :-)

编辑 1: 正如viakondratiuk在if n is 1中提到的,它应该从0开始,而不是1.

编辑 2:

您可以使用 ** 符号来避免将 float 转换为 int。

>>> n = 2
>>> range(10 ** (n - 1), 10 ** n)
[10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99]

您可以将 powrange 结合使用(在 Python 2.7 中,使用 xrange,这样可以避免在内存中分配整个列表):

n = 1
start = pow(10, n)
r = (range(10) if n == 1 else range(start, start * 10))
[i for i in r]
# [1, 2, 3, 4, 5, 6, 7, 8, 9]