如何迭代熊猫数据框并根据第三列比较某些列？

Question

我是 pandas 的新手，很难以方便的方式使用它的功能。

我有一个大型数据框，其中包含我想比较的两个不同测试的实验数据。理想情况下，数据显示在图中。

## what I have:
import pandas as pd

ids = [
    'Bob','Bob',
    'John', 'John',
    'Mary', 'Mary',
    ]
var = [
    'a', 'b',
    'a', 'b',
    'a', 'b',
    ]
data = [
    10,11,
    15,14,
    10,15
    ]
dataset = zip(ids, var, data)
print dataset

columns = ['ids', 'var', 'data']
df = pd.DataFrame(data = dataset, columns=columns)
print df

## what I want:
import matplotlib.pyplot as plt
from matplotlib.ticker import MultipleLocator

fig = plt.figure()
ax1 = fig.add_subplot(111)
for i,ii in enumerate(ids):
    if var[i] == 'a':
        ax1.plot(i/2, data[i], 'rs', label='var a')
    else:
        ax1.plot((i-1)/2, data[i], 'bo', label='var b')
majorLocator = MultipleLocator(1)
ax1.xaxis.set_major_locator(majorLocator)
ax1.grid()
ax1.margins(0.05)
ax1.set_xlabel('ids')
ax1.set_ylabel('data')
ax1.legend(loc='best', numpoints=1)
fig.show()

在没有很多嵌套 for 循环的情况下如何正确执行此操作？如果我可以将 id 用作 xlabels，那将是一个加分项......

提前非常感谢， 丹尼尔

Answer 1

我不太确定您想要的最终目标是什么，但如果 cphlewis 建议使用 seaborn 不是您想要的，您可以尝试将 DataFrame 转换为多索引，然后将其绘制出来那样。

mi = pd.DataFrame(data=data,index=[ids,var],columns=['data'])
f, a = plt.subplots()
mi.plot(kind='bar',ax=a)

参考this post 也可能会有所帮助。

Answer 2

seaborn 为你做了很多，非常灵活：

import seaborn as sns
sns.factorplot('ids', 'data', hue='var', kind='bar', data=df)

（它还重新设置了绘图默认值，可以更改或重置）。

如果要对数据进行子集化，请将子集作为 data 参数传递：

sns.factorplot('ids', 'data', hue='var', kind='bar', 
               data=df[df.isin({'ids':['Bob','Mary']}).any(1)])

• sns 样式已关闭
• 对于更复杂的掩码，您需要单独设置掩码；查看熊猫文档