首席鉴黄大师
问题描述
我有一些数据存储为带有 dtype=object 的 numpy 数组，我想提取一列列表并将其转换为 numpy 数组.这似乎是一个简单的问题，但我发现解决它的唯一方法是将整个事物重铸为列表列表，然后将其重铸为 numpy 数组.有没有更 pythonic 的方法?
i have some data which is stored as a numpy array with dtype=object, and i would like to extract one column of lists and convert it to a numpy array. it seems like a simple problem, but the only way i've found to solve it is to recast the entire thing as a list of lists and then recast it as a numpy array. is there a more pythonic approach?
import numpy as np arr = np.array([[1, ['a', 'b', 'c']], [2, ['a', 'b', 'c']]], dtype=object) arr = arr[:, 1] print(arr) # [['a', 'b', 'c'] ['a', 'b', 'c']] type(arr) # numpy.ndarray type(arr[0]) # list arr.shape # (2,)
将数组重铸为 dtype=str 会引发 valueerror，因为它试图将每个列表转换为字符串.
recasting the array as dtype=str raises a valueerror since it is trying to convert each list to a string.
arr.astype(str) # valueerror: setting an array element with a sequence
可以将整个数组重建为列表列表，然后将其转换为 numpy 数组，但这似乎是一种迂回的方式.
it is possible to rebuild the entire array as a list of lists and then cast it as a numpy array, but this seems like a roundabout way.
arr_2 = np.array(list(arr)) type(arr_2) # numpy.ndarray type(arr_2[0]) # numpy.ndarray arr_2.shape # (2, 3)
有没有更好的方法来做到这一点?
is there a better way to do this?
推荐答案
虽然通过列表的方式比通过 vstack 的方式更快:
though going by way of lists is faster than by way of vstack:
in [1617]: timeit np.array(arr[:,1].tolist()) ... 100000 loops, best of 3: 11.5 μs per loop in [1618]: timeit np.vstack(arr[:,1]) ... 10000 loops, best of 3: 54.1 μs per loop
vstack 正在做:
np.concatenate([np.atleast_2d(a) for a in arr[:,1]],axis=0)
一些替代方案:
in [1627]: timeit np.array([a for a in arr[:,1]]) 100000 loops, best of 3: 18.6 μs per loop in [1629]: timeit np.stack(arr[:,1],axis=0) 10000 loops, best of 3: 60.2 μs per loop
请记住，对象数组只包含指向内存中其他位置的列表的指针.虽然 arr 的 2d 特性使得选择第二列变得容易，但 arr[:,1] 实际上是一个列表列表.并且对其进行的大多数操作都是这样对待的.reshape 之类的东西不会跨越 object 边界.
keep in mind that the object array just contains pointers to the lists which are else where in memory. while the 2d nature of arr makes it easy to select the 2nd column, arr[:,1] is effectively a list of lists. and most operations on it treat it as such. things like reshape don't cross that object boundary.
