我如何将熊猫系列或索引转换为NumPy数组?

我如何能得到一个数据帧作为NumPy数组或Python列表的索引或列?

565592 次浏览

可以使用df.index访问索引对象,然后使用df.index.tolist()获取列表中的值。类似地,你可以为Series使用df['col'].tolist()

要获得NumPy数组,你应该使用values属性:

In [1]: df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]}, index=['a', 'b', 'c']); df
A  B
a  1  4
b  2  5
c  3  6


In [2]: df.index.values
Out[2]: array(['a', 'b', 'c'], dtype=object)

这将访问数据已经存储的方式,因此不需要进行任何转换。

注意:此属性也可用于许多其他pandas对象。

In [3]: df['A'].values
Out[3]: Out[16]: array([1, 2, 3])

要获取列表形式的索引,调用tolist:

In [4]: df.index.tolist()
Out[4]: ['a', 'b', 'c']

类似地,对于列。

从pandas v0.13开始,你还可以使用get_values:

df.index.get_values()

如果您正在处理一个多索引数据框架,您可能只对提取多索引的一个名称的列感兴趣。你可以这样做

df.index.get_level_values('name_sub_index')

当然,name_sub_index必须是FrozenList df.index.names的元素

我将熊猫dataframe转换为list,然后使用基本的list.index()。就像这样:

dd = list(zone[0]) #Where zone[0] is some specific column of the table
idx = dd.index(filename[i])

索引值为idx

熊猫>= 0.24

弃用.values,转而使用这些方法!

从v0.24.0开始,我们将有两个全新的首选方法来从IndexSeriesDataFrame对象中获取NumPy数组:它们是to_numpy().array。关于用法,文档中提到:

我们还没有删除或弃用Series.valuesDataFrame.values,但,我们强烈建议使用.array.to_numpy()。< /强> < / p >

更多信息请参见v0.24.0版本的本节说明


to_numpy() Method . sh DataFrames(而.array则不行)。


< p > array属性 . rref ="https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.Series.array.html#pandas.Series.array" rel="nofollow noreferrer">array
此属性返回一个支持Index/Series的ExtensionArray对象
pd.__version__
# '0.24.0rc1'


# Setup.
df = pd.DataFrame([[1, 2], [4, 5]], columns=['A', 'B'], index=['a', 'b'])
df


A  B
a  1  2
b  4  5

& lt; !- - - - - -→

df.index.array
# <PandasArray>
# ['a', 'b']
# Length: 2, dtype: object


df['A'].array
# <PandasArray>
# [1, 4]
# Length: 2, dtype: int64

从这里,可以使用list获取一个列表:

list(df.index.array)
# ['a', 'b']


list(df['A'].array)
# [1, 4]

或者,直接调用.tolist():

df.index.tolist()
# ['a', 'b']


df['A'].tolist()
# [1, 4]

关于返回什么,文档提到,

对于普通NumPy数组支持的SeriesIndexes, Series.array . bb0

将返回一个新的arrays.PandasArray,它是一个thin(无复制) numpy.ndarray的包装器。arrays.PandasArray不是特别的 它本身很有用,但它确实提供了与其他工具相同的接口 在pandas中或由第三方库定义的扩展数组

因此,概括地说,.array将返回任意一个

  1. 现有的ExtensionArray支持索引/系列,或
  2. 如果有一个NumPy数组支持该系列,则会创建一个新的ExtensionArray对象作为底层数组的精简包装器。

< p > 添加两个新方法的基本原理
这些函数是在两个GitHub问题GH19954GH23623下讨论的结果

具体来说,文档中提到了基本原理:

< p >[….values,不清楚返回值是否为 实际的阵列,它的一些变换,或熊猫的习俗之一 数组(如Categorical)。例如,使用PeriodIndex.values 每次生成一个新的周期对象ndarray。[…]< / p >

这两个函数旨在提高API的一致性,这是朝着正确方向迈出的重要一步。

最后,.values在当前版本中不会被弃用,但我预计这可能会在未来的某个时候发生,所以我会敦促用户尽快迁移到更新的API。

下面是一个将数据帧列转换为NumPy数组的简单方法。

df = pd.DataFrame(somedict)
ytrain = df['label']
ytrain_numpy = np.array([x for x in ytrain['label']])

ytrain_numpy是一个NumPy数组。

我尝试了to.numpy(),但它给了我以下错误:

TypeError:不支持类型转换:(dtype('O'),)*,同时使用线性SVC进行二进制相关性分类。

to.numpy()正在将dataFrame转换为NumPy数组,但内部元素的数据类型是列表,因此观察到上述错误。

最近的一种方法是使用.to_numpy()函数。

如果我有一个列为“price”的数据框架,我可以将其转换为如下方式:

priceArray = df['price'].to_numpy()

还可以将数据类型(如float或object)作为函数的参数传递