首页 > 后端开发 > Python教程 > 如何从 Pandas DataFrame 中提取数据作为 Python 列表?

如何从 Pandas DataFrame 中提取数据作为 Python 列表?

Barbara Streisand
发布: 2024-10-26 16:00:04
原创
731 人浏览过

How can I extract data from a Pandas DataFrame as a Python list?

从 Pandas DataFrame 访问数据

要从 Pandas DataFrame 中检索数据作为列表,有多种方法可用,具体取决于您是否想要访问特定的列或行。

以列表形式访问列

要以列表形式获取列的内容,可以使用以下语法:

<code class="python">column_list = df['column_name'].tolist()</code>
登录后复制

这将返回一个包含指定列的元素的 Python 列表。

将行作为列表访问

检索一行作为列表,您可以使用 iloc 或 loc 方法,后跟 .tolist():

<code class="python">row_list = df.iloc[row_index].tolist()  # row index-based
row_list = df.loc[row_label].tolist()  # row label-based</code>
登录后复制

示例

考虑 DataFrame:

<code class="python">import pandas as pd

df = pd.DataFrame({'cluster': ['A', 'A', 'A', 'B', 'B', 'B', 'C', 'C', 'C'],
                   'load_date': ['1/1/2014', '2/1/2014', '3/1/2014', '4/1/2014', '4/1/2014', '4/1/2014', '7/1/2014', '8/1/2014', '9/1/2014'],
                   'budget': [1000, 12000, 36000, 15000, 12000, 90000, 22000, 30000, 53000],
                   'actual': [4000, 10000, 2000, 10000, 11500, 11000, 18000, 28960, 51200],
                   'fixed_price': ['Y', 'Y', 'Y', 'N', 'N', 'N', 'N', 'N', 'N']})</code>
登录后复制

以列表形式访问列:

<code class="python">cluster_list = df['cluster'].tolist()  # ['A', 'A', 'A', 'B', 'B', 'B', 'C', 'C', 'C']</code>
登录后复制

以列表形式访问行:

<code class="python">row_list = df.iloc[0].tolist()  # ['A', '1/1/2014', 1000, 4000, 'Y']</code>
登录后复制

以上是如何从 Pandas DataFrame 中提取数据作为 Python 列表?的详细内容。更多信息请关注PHP中文网其他相关文章!

来源:php.cn
本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
作者最新文章
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板