温馨提示×

pandas怎么获取某一列数据

小亿
400
2023-12-29 09:45:25
栏目: 编程语言

要获取Pandas DataFrame 中的某一列数据,可以使用以下两种方法:

  1. 使用 DataFrame[column]:可以通过列名来获取某一列的数据。例如,如果 DataFrame 中有一列名为 “column_name”,可以使用 DataFrame['column_name'] 来获取该列的数据。

  2. 使用 DataFrame.column:如果 DataFrame 中的列名是有效的 Python 变量名称,可以使用该列名来获取该列的数据。例如,如果 DataFrame 中有一列名为 “column_name”,可以使用 DataFrame.column_name 来获取该列的数据。

以下是一个示例代码,演示如何获取 DataFrame 中的某一列数据:

import pandas as pd

# 创建一个示例 DataFrame
data = {'column1': [1, 2, 3, 4, 5],
        'column2': ['a', 'b', 'c', 'd', 'e']}
df = pd.DataFrame(data)

# 使用 DataFrame[column] 获取某一列数据
column1_data = df['column1']
print(column1_data)

# 使用 DataFrame.column 获取某一列数据
column2_data = df.column2
print(column2_data)

以上示例中,df['column1']df.column2 分别获取了 DataFrame df 中的 ‘column1’ 和 ‘column2’ 列的数据。

0