3

我正在做一些太阳系动力学模拟,并且一直以这个项目为借口自学一些 python/pandas。生成的数据集有超过 1000 条记录,其中包含每个涉及的天体的轨道倾角、偏心率等值。

我正在尝试使用 pandas 来研究结果数据中约 1000 个物体(“测试粒子”)的平均轨道倾角,作为半长轴的函数。

到目前为止,我一直在做的是:

1)将数据读入数据帧

df = pd.read_csv('final.csv')

2)将数据限制在半长轴值的范围内(粒子在模拟过程中“扩散”,但我想限制我的分析

cf = df[df.a.between(30,80)]

3) 绘制给定箱数的倾斜度平均值

cf.groupby(pd.cut(cf.a, 80))['inc'].mean().plot()

这会创建一个可接受的绘图,但在格式方面它有几个问题:除非它完全最大化以填满我的屏幕,否则沿 x 轴的数字会被挤压在一起并重叠。它们也不完全是我想看到的:它们显示了垃圾箱的最大值和最小值,我更喜欢直接按 5s 或类似的方式滴答作响。

上述命令的熊猫输出

我尝试将值传递(x=None, xticks=None)plot()上面的字符串部分,但这对结果图没有影响。我这样做的方式是否可以进行绘图/输出控制?

谢谢,

G

4

1 回答 1

6

当我想做这样的事情时,我直接去matplotlib。我将用这个样本数据展示一个小例子:

df = pd.DataFrame([[1, 2], [2, 7], [3, 6], [4,7], [5,3]], columns=['A', 'B'])

pd.cut我没有用 切割它,而是用 切割它np.linspace。所以

bins = np.linspace(0,5,4)
group = df.groupby(pd.cut(df.A, bins))

现在,要绘制它,我想要垃圾箱的中间

plot_centers = (bins [:-1] + bins [1:])/2
plot_values = group.B.mean()

并与

plt.plot(plot_centers, plot_values)

您应该小心处理丢失的数据,即如果您有一个没有数据的 bin。在这种情况下,您可以使用fillna(0), 使全NaNs为 0。

plot_values = group.B.mean().fillna(0)
于 2017-11-07T14:11:45.380 回答