从 np.arrays

问题描述

我有一个数组列表,我想将每个数组的前 10 列传递给一个缩放器来转换它们,但不传递其余的列,因为它们是虚拟变量。

每个单独的数组都是二维的,并包含与特定列对应的数据。

我试过了:

list[:][:10]

但这只是给了我前 10 个数组,而不是所有数组的前 10 列。

解决方法

您也可以将外部列表转换为 numpy 数组 (np.array(my_list)) 然后使用多维索引,如:my_np_list[:,:10]

,

假设每个数组都具有相同的长度,尝试使用 np.stack 进行索引 -

arr = [np.array([1,2,3,4]),np.array([4,5,6,7]),np.array([8,9,10,11])]

#Getting the first 2 column (:10 for first 10)
np.stack(arr)[:,:2]
array([[1,2],[4,5],[8,9]])