我需要在列的0级绘制带有ylabels的MultiIndex数据帧。 Dataframe如下:
PCs pc1 pc2 pc3 \
explained_variance_ratio 9.977643e-01 2.196399e-03 3.275875e-05
wavelength
540.00 0.015110 -0.004772 -0.018467
540.05 0.015110 -0.004772 -0.018467
540.10 0.015110 -0.004772 -0.018467
540.15 0.015110 -0.004772 -0.018467
540.20 0.015110 -0.004772 -0.018467
540.25 0.015110 -0.004772 -0.018467
540.30 0.015110 -0.004772 -0.018467
540.35 0.015110 -0.004772 -0.018467
540.40 0.015081 -0.004226 -0.017577
540.45 0.015081 -0.004226 -0.017577
.......
ylabel应该是pc1,pc2,pc3 ....波长为xlabel。
我尝试了以下它不起作用:
pca_df.plot(subplots=True, sharex=True, title='Principle Components', \
legend=False, y=list(pca_df.columns.levels[0]))
错误是:
Traceback (most recent call last):
File "<stdin>", line 2, in <module>
File "/usr/lib/python2.7/dist-packages/pandas/tools/plotting.py", line 1716, in plot_frame
ser = frame[y]
File "/usr/lib/python2.7/dist-packages/pandas/core/frame.py", line 1652, in __getitem__
return self._getitem_array(key)
File "/usr/lib/python2.7/dist-packages/pandas/core/frame.py", line 1697, in _getitem_array
return self.take(indexer, axis=1, convert=True)
File "/usr/lib/python2.7/dist-packages/pandas/core/generic.py", line 1166, in take
indices, len(self._get_axis(axis)))
File "/usr/lib/python2.7/dist-packages/pandas/core/indexing.py", line 1438, in _maybe_convert_indices
if mask.any():
AttributeError: 'bool' object has no attribute 'any'
也尝试过:
pca_df.plot(subplots=True, sharex=True, title='Principle Components', \
legend=False)
plt.ylabel(pca_df.columns.levels[0])
虽然它描绘了覆盖所有情节的ylabel,但是:
Index([u'pc1', u'pc10', u'pc2', u'pc3', u'pc4', u'pc5', u'pc6', u'pc7', u'pc8', u'pc9'], dtype='object')
答案 0 :(得分:3)
在调用plot
后,我通常会对斧头或轴本身进行大部分格式化。
试试这个:
axes = pca_df.plot(subplots=True, sharex=True, title='Principle Components', legend=False)
for ax, label in zip(axes.ravel(), pca_df.columns.levels[0]):
ax.set_ylabel(label)
我发现使用轴API更容易,而不是试图将其全部放入绘图方法调用中。请注意,当子图为True时,plot方法返回一个轴数组,您必须通过索引或迭代来访问这些轴,如我在此处所做的那样。 .ravel
是处理二维数组的一种很好的方法,当你的数字有多个行和列时,你会得到它,但如果数组只有一维,它仍然可以工作。