绘制在散点图中的pandas数据框中分组的数据

Question

我有大量的天文数据需要在散点图中绘制。我根据距离对数据进行了分类，我想并排绘制4个散点图。

为了提出这个问题，我构建了一个基于MWE，显然有不同的数据，我到目前为止所得到的：

import pandas as pd
import matplotlib.pyplot as plt

data = {'Name':['Tom', 'Jack', 'Steve', 'Ricky', 'Jim', 'Lee', 'Rob', 'Dave',
                'Jane', 'Bronwyn', 'Karen', 'Liz', 'Claire', 'Chris', 'Jan', 'Ruby'],
         'Age':[28,34,29,42,14,16,75,68,
                27,3,2,19,17,32,71,45],
         'Weight':[60,75,73,82,54,55,98,82,45,9,8,47,54,62,67,67]}

stages = ['Toddler', 'Teen', ' Young Adult', 'Adult']
ages = [0,4,20,40,100]
df = pd.DataFrame(data)

df['binned'] = pd.cut(df['Age'], bins=ages, labels=stages)

fig=plt.figure()
fig.subplots_adjust(hspace=0) 
fig.subplots_adjust(wspace=0) 
gridsize = 1,4

ax1 = plt.subplot2grid(gridsize, (0,0))
ax1.scatter(df['Name'], df['Weight'], alpha = 0.5)
ax1.set_ylabel('Weight, kg', fontsize=20)
ax1.set_xlabel('Name', fontsize=20)

ax2 = plt.subplot2grid(gridsize, (0,1), sharey=ax1, sharex = ax1)
plt.setp(ax2.get_yticklabels(), visible=False)
ax2.scatter(df['Name'], df['Weight'], alpha = 0.5)
ax2.set_xlabel('Name', fontsize=20)

ax3 = plt.subplot2grid(gridsize, (0,2), sharey=ax1, sharex = ax1)
plt.setp(ax3.get_yticklabels(), visible=False)
ax3.scatter(df['Name'], df['Weight'], alpha = 0.5)
ax3.set_xlabel('Name', fontsize=20)

ax4 = plt.subplot2grid(gridsize, (0,3), sharey=ax1, sharex = ax1)
plt.setp(ax4.get_yticklabels(), visible=False)
ax4.scatter(df['Name'], df['Weight'], alpha = 0.5)
ax4.set_xlabel('Name', fontsize=20)

这绘制了预期的四个图：但是如何让每个图只绘制每个箱中一个的数据？换句话说，我如何只绘制其中一个箱子？

我并不担心x轴上名称的缩小，这些只是针对这个MWE。他们将是我实际情节中的数字。

只是为了澄清，我的实际数据被分类为

sources['z bins']=pd.cut(sources['z'], [0,1,2,3, max(z)],
                   labels = ['z < 1', '1 < z < 2', '2 < z < 3', 'z > 3'])

Answer 1

如果您按binned对数据框进行分组然后绘制每个组，该怎么办？

例如：

fig=plt.figure()
fig.subplots_adjust(hspace=0) 
fig.subplots_adjust(wspace=0) 
gridsize = 1,4

for i, (name, frame) in enumerate(df.groupby('binned')):
    ax = plt.subplot2grid(gridsize, (0,i))
    ax.scatter(frame['Name'], frame['Weight'], alpha = 0.5)
    ax.set_xlabel(name, fontsize=20)

我意识到你可能想稍微清理标签，但这至少会将不同的箱子放在不同的轴对象上。

您可以迭代groupby对象并返回组的名称和该组的数据框。这里我使用枚举来增加轴对象

或者，如果您不想使用for循环，则可以使用groupby对象的get_group方法访问每个组。

grouped = df.groupby('binned')

ax1 = plt.subplot2grid(gridsize, (0,0))

ax1.scatter(grouped.get_group('Toddler')['Name'],
            grouped.get_group('Toddler')['Weight'],
            alpha = 0.5)
ax1.set_ylabel('Weight, kg', fontsize=20)
ax1.set_xlabel('Name', fontsize=20)