如何将 Pandas 系列或索引转换为 NumPy 数组?[重复]

2025-02-05 13:23:00
admin
原创
74
摘要:问题描述:如何将 DataFrame 的索引或列作为 NumPy 数组或 Python 列表获取?解决方案 1:要获取 NumPy 数组,您应该使用以下values属性:In [1]: df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]}, index=['a'...

问题描述:

如何将 DataFrame 的索引或列作为 NumPy 数组或 Python 列表获取?


解决方案 1:

要获取 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']

对于列来说也是类似。

解决方案 2:

您可以使用df.index访问索引对象,然后使用 获取列表中的值df.index.tolist()。同样,您也可以使用df['col'].tolist()Series。

解决方案 3:

熊猫> = 0.24

反对使用而.values赞成使用这些方法!

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

我们并没有删除或弃用Series.values
DataFrame.values,但**我们强烈建议使用.array
.to_numpy()代替。**

有关更多信息,请参阅v0.24.0 发行说明的此部分。


to_numpy()方法

df.index.to_numpy()
# array(['a', 'b'], dtype=object)

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

默认情况下,返回一个视图。所做的任何修改都会影响原始视图。

v = df.index.to_numpy()
v[0] = -1
 
df
    A  B
-1  1  2
b   4  5

如果您需要副本,请使用to_numpy(copy=True);

v = df.index.to_numpy(copy=True)
v[-1] = -123
 
df
   A  B
a  1  2
b  4  5

请注意,此函数也适用于DataFrames(但.array不适用)。


array属性

此属性返回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

<!- ->

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 数组支持的Series和,
将返回一个新的,它是 的薄(无复制)包装器。它本身并不是特别有用,但它确实提供了与 pandas 或第三方库中定义的任何扩展数组相同的接口。Index`Series.arrayarrays.PandasArraynumpy.ndarray`arrays.PandasArray

因此,总结一下,.array将返回

  1. 现有的ExtensionArray支持指数/系列的资产,或

  2. 如果有一个 NumPy 数组支持该系列,ExtensionArray则会创建一个新对象作为底层数组的薄包装器。


添加两种新方法的理由这两个功能是根据 GitHub 上两个问题GH19954和GH23623

的讨论结果添加的。

具体来说,文档提到了其理由:

[...].values不清楚返回的值是实际数组、数组的某种变换,还是 pandas 自定义数组之一(如Categorical)。例如,使用PeriodIndex,每次.values
都会生成一个新的period 对象。[...]ndarray

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

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

解决方案 4:

如果您正在处理多索引数据框,您可能只想提取多索引中一个名称的列。您可以这样做

df.index.get_level_values('name_sub_index')

当然name_sub_index必须是FrozenList df.index.names

解决方案 5:

从 pandas v0.13 开始你也可以使用get_values

df.index.get_values()

解决方案 6:

较新的方法是使用 .to_numpy() 函数。

如果我有一个带有“价格”列的数据框,我可以按如下方式转换它:

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

您还可以将数据类型(例如浮点型或对象)作为函数的参数传递

解决方案 7:

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

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

您的索引值为idx

解决方案 8:

以下是将数据框列转换为 NumPy 数组的简单方法。

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

ytrain_numpy是一个 NumPy 数组。

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

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

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

相关推荐
  政府信创国产化的10大政策解读一、信创国产化的背景与意义信创国产化,即信息技术应用创新国产化,是当前中国信息技术领域的一个重要发展方向。其核心在于通过自主研发和创新,实现信息技术应用的自主可控,减少对外部技术的依赖,并规避潜在的技术制裁和风险。随着全球信息技术竞争的加剧,以及某些国家对中国在科技领域的打压,信创国产化显...
工程项目管理   1590  
  为什么项目管理通常仍然耗时且低效?您是否还在反复更新电子表格、淹没在便利贴中并参加每周更新会议?这确实是耗费时间和精力。借助软件工具的帮助,您可以一目了然地全面了解您的项目。如今,国内外有足够多优秀的项目管理软件可以帮助您掌控每个项目。什么是项目管理软件?项目管理软件是广泛行业用于项目规划、资源分配和调度的软件。它使项...
项目管理软件   1361  
  信创产品在政府采购中的占比分析随着信息技术的飞速发展以及国家对信息安全重视程度的不断提高,信创产业应运而生并迅速崛起。信创,即信息技术应用创新,旨在实现信息技术领域的自主可控,减少对国外技术的依赖,保障国家信息安全。政府采购作为推动信创产业发展的重要力量,其对信创产品的采购占比情况备受关注。这不仅关系到信创产业的发展前...
信创和国产化的区别   18  
  信创,即信息技术应用创新产业,旨在实现信息技术领域的自主可控,摆脱对国外技术的依赖。近年来,国货国用信创发展势头迅猛,在诸多领域取得了显著成果。这一发展趋势对科技创新产生了深远的推动作用,不仅提升了我国在信息技术领域的自主创新能力,还为经济社会的数字化转型提供了坚实支撑。信创推动核心技术突破信创产业的发展促使企业和科研...
信创工作   18  
  信创技术,即信息技术应用创新产业,旨在实现信息技术领域的自主可控与安全可靠。近年来,信创技术发展迅猛,对中小企业产生了深远的影响,带来了诸多不可忽视的价值。在数字化转型的浪潮中,中小企业面临着激烈的市场竞争和复杂多变的环境,信创技术的出现为它们提供了新的发展机遇和支撑。信创技术对中小企业的影响技术架构变革信创技术促使中...
信创国产化   19  
热门文章
项目管理软件有哪些?
云禅道AD
禅道项目管理软件

云端的项目管理软件

尊享禅道项目软件收费版功能

无需维护,随时随地协同办公

内置subversion和git源码管理

每天备份,随时转为私有部署

免费试用