有人可以使用 numpy vstack 帮助我处理“for 循环”吗？

Question

假设我有大约 640 个 numpy 数组要垂直堆叠。每个数组的大小为 (66, 1)。像这样手动执行此操作：

A = np.vstack((Ne['State_1_inc'], Ne['State_2_inc'], Ne['State_3_inc'], Ne['State_4_inc'], ..., Ne['State_640_inc']))

显然会花费很长时间，而且非常耗时。最终结果的大小为 A (66,640)。任何人都知道我是否可以做一个 for loop 来传递我所有的 640 个状态，以便我可以构建我的矩阵？编程新手，谢谢！

Answer 1

假设您想使用字典中的所有元素：

Ne = {1: [1,2,3], 2: [4,5,6]}
np.vstack(list(Ne.values()))
# array([[1, 2, 3],
#        [4, 5, 6]])

否则你可以使用字典理解：

np.vstack([Nef[f'State_{i+1}_inc'] for i in range(640)])

Can anybody help me with my `for loops` using numpy vstack?