假設我有大約 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 個狀態的方法,以便我可以構建我的矩陣?編程新手,謝謝!
假設您想使用字典的所有元素:
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)])
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句