python中的for循环一行

2024-06-06 05:13:59 发布

您现在位置:Python中文网/ 问答频道 /正文

当我试图用下面这样的一行来陈述时

def get_cubes(x):
ls=[]
ls.append(pow(item*3, 3)) for item in range(int((x-x%3)/3)+1)
return ls

总是有一个错误:

File "<ipython-input-47-8c391c3d568a>", line 3
ls.append(pow(item*3, 3))  for item in range(int((x-x%3)/3)+1)
                             ^
SyntaxError: invalid syntax

但当我用常规for循环方法编写它时,它工作得很好。有人能告诉我为什么吗?


Tags: inforgetreturndef错误rangeitem
3条回答

解决方案

如果您严格要求一行,那么这就是解决方案:

get_cubes = lambda x: [pow(i, 3) for i in range(0, x+1, 3)]

但是由于在Python中,清晰性和可读性应该始终是优先考虑的,所以最好像@daniel那样将其作为函数编写:

def get_cubes(x):
    return [pow(i, 3) for i in range(0, x+1, 3)]

输出

使用for循环测试函数可以得到以下结果:

for i in range(20):
    print(i, ':', get_cubes(i))

0 : [0]
1 : [0]
2 : [0]
3 : [0, 27]
4 : [0, 27]
5 : [0, 27]
6 : [0, 27, 216]
7 : [0, 27, 216]
8 : [0, 27, 216]
9 : [0, 27, 216, 729]
10 : [0, 27, 216, 729]
11 : [0, 27, 216, 729]
12 : [0, 27, 216, 729, 1728]
13 : [0, 27, 216, 729, 1728]
14 : [0, 27, 216, 729, 1728]
15 : [0, 27, 216, 729, 1728, 3375]
16 : [0, 27, 216, 729, 1728, 3375]
17 : [0, 27, 216, 729, 1728, 3375]
18 : [0, 27, 216, 729, 1728, 3375, 5832]
19 : [0, 27, 216, 729, 1728, 3375, 5832]

解释

对于那些问为什么@daniel的代码有效的人:

原始代码执行以下操作:

  1. 给定一个x,从0迭代到3 divides x加上1的次数。
  2. 对于每个x,乘以3,并将其提升到3的幂次
  3. 返回包含步骤2结果的列表

步骤1最初编写为

range(int((x - x%3) / 3) + 1)

当除以3时减去一个数字的提醒,然后在加1之前除以3。但是同样的结果也可以通过把x除以3的整数部分,然后再加上1得到,这看起来像这样:

int(x / 3) + 1

第2步将每个迭代乘以3,然后将其提升到3的幂次,但也可以通过使用范围(就像@daniel所做的那样)来实现“乘法”,方法是:

range(0, x+1, 3)

0x+13的步骤上迭代,得到相同的结果。一、 ex正在10

range(0, 10 + 1, 3)  |    i*3 for i in range(int(x / 3) + 1)
============================================================
0                    |    0*3 = 0
3                    |    1*3 = 3
6                    |    2*3 = 6
9                    |    3*3 = 9

对于步骤3,我们只需要应用pow(x, 3)(或x ** 3),然后使用列表理解将所有内容都放在一行中,然后我们可以将其放在lambda函数中或函数的return语句中。

[pow(i, 3) for i in range(0, x+1, 3)]

我的回答遵循(修改)您的方法(代码),而不是使用列表理解,而是以正确的方式。将它与您的代码进行比较,您将找出错误所在。我添加了注释#来解释更改

def get_cubes(x):
    ls=[] # create an empty list outside the for loop once
    for item in range(int((x-x%3)/3)+1): # for loop for iterating through elements
        ls.append(pow(item*3, 3)) # append one item at a time
    return ls

print (get_cubes(10))                                   
> [0, 27, 216, 729]

for循环的正确语法是

def get_cubes(x):
    ls = []
    for item in range(int((x-x%3)/3)+1):
        ls.append(pow(item*3, 3)) 
    return ls

也许,当循环体仅附加到列表时,您正在寻找列表理解,这是一种生成列表的方法:

def get_cubes(x):
    ls = [(item*3) ** 3 for item in range(int((x-x%3)/3)+1)]
    return ls

或者简而言之:

def get_cubes(x):
    return [i ** 3 for i in range(0, x+1, 3)]

相关问题 更多 >