Question

我有一个函数，可以在散点图上绘制Pandas列。我想使用列的名称作为轴标签。这是我的意思的一个例子：

def plotscatter(x,y):
    plt.scatter(x, y)
    plt.xlabel(x)
    plt.ylabel(y)
plotscatter(df['1stcol'],df['2ndcol'])

我想提取引号中的文本，即1stcol和2ndcol。如果我按原样使用x和y，我会得到列名，后跟所有值作为轴标签。

是否有内置方法或功能？像x.columnname（）

Answer 1

您可以使用DataFrame.columns或列Series.name访问列名称。一个例子：

In [11]: df = pd.DataFrame([[1,2],[2,4]], columns=['1stcol', '2ndcol'])

In [13]: df.columns
Out[13]: Index([u'1stcol', u'2ndcol'], dtype='object')

In [14]: df.columns[0]
Out[14]: '1stcol'

In [16]: df['1stcol'].name
Out[16]: '1stcol'

从Pandas列索引中提取列名

1 个答案: