Pandas获取dataframe列名

最新推荐文章于 2025-06-12 11:37:08 发布

背生双翼

最新推荐文章于 2025-06-12 11:37:08 发布

阅读量572

点赞数 2

CC 4.0 BY-SA版权

文章标签： pandas

本文链接：https://blog.csdn.net/zhtyok/article/details/141066101

利用data.columns获取列名

import pandas as pd
data = pd.DataFrame({
    'A': [1, 2, 3, 4],
    'B': [5, 6, 7, 8],
    'C': [9, 10, 11, 12]})#新建 dataframe
column_names = data.columns#获取列名
print(column_names)#输出列名，['A', 'B', 'C']

获取指定列

import pandas as pd
data = pd.DataFrame({
    'A': [1, 2, 3, 4],
    'B': [5, 6, 7, 8],
    'C': [9, 10, 11, 12]})#新建 dataframe
Num = ['A','B']#需要获取的列
df = data[Num] #获取指定列
print(df)#打印dataframe