能否用单条语句修改NumPy数组元素并返回数组？

当然可以实现！你遇到的问题核心在于：itemset() 和直接的赋值操作（比如 a[x,0] +=1）都是原地修改NumPy数组，它们不会返回修改后的数组本身，而是返回None——这就是为什么你的列表推导式最后得到的是一堆None，而不是预期的数组列表。

下面给你两种解决方案，一种是通用的“修改后返回数组”写法，另一种是针对你这个具体场景的更简洁优化方案：

方案1：通用场景——原地修改后返回数组（Python 3.8+） #

利用Python 3.8引入的**海象运算符（:=）**先把数组赋值给变量，执行修改操作后，再返回这个变量。因为修改操作返回None，我们可以用or运算符来取到后面的数组变量：

import numpy as np
arr = [ (a := np.zeros((5, 1)))[x, 0] += 1 or a for x in range(5) ]

如果需要做普通赋值（不是增强赋值），可以调用数组的__setitem__方法来完成修改，同样结合海象运算符：

arr = [ (a := np.zeros((5, 1))).__setitem__((x, 0), 1) or a for x in range(5) ]

方案2：针对你的具体场景——直接生成目标数组（更高效简洁） #

观察你的需求：每个数组是(5,1)的形状，只有第x行是1，其余为0——这不就是单位矩阵的列向量吗？用np.eye可以直接生成，完全不需要先创建zeros再修改：

import numpy as np
arr = [ np.eye(5)[x].reshape(-1, 1) for x in range(5) ]

或者更直观的切片写法：

arr = [ np.eye(5)[[x], :].T for x in range(5) ]

这种方法不仅代码更短，而且避免了原地修改的操作，效率也更高。

内容的提问来源于stack exchange，提问作者JustAGuy