创建考虑三个数据集的箱

时间:2019-04-16 11:03:58

标签: python numpy matplotlib statistics binning

我有三个角度:以二维数组描述的Rot,Tilt和Psi,其中每行描述每个数据点的Rot,Tilt和Psi角度

我想分析这些角度以查看何时有数据显示我是否在特定的欧拉角上有大量数据,然后使用它来修整数据集,以便每个后我有更多偶数个粒子斌

当前,我已使用hist2D绘制了数据,虽然从热图上可以看到,但我有一些区域具有更多的数据点,而我认为如果不在3D模式下看不到它,我将无法看到整个图像。

import pandas as pd
from mpl_toolkits import mplot3d
from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
import numpy as np

def Hist2D(X,Y,Z,Bins):
    fig, (XY, XZ, YZ) = plt.subplots(1, 3)
    XY.hist2d(X,Y,Bins)
    XY.set_title('AngleRot - AngleTilt')
    XZ.hist2d(X,Z,Bins)
    XZ.set_title('AngleRot - AnglePsi')
    YZ.hist2d(Y,Z,Bins)
    YZ.set_title('AngleTilt - AnglePsi')
    plt.show()

#For the purposes of this example I have just given a bunch of small 1D arrays.
Tilt = [120,90,40,40]
Rot = [90,20,3,56]
Psi = [5, 67, 3, 4]

Hist2D(Rot, Tilt, Psi, 10)

这为我提供了2D直方图,但如果可能的话,我希望获得3D直方图。

有没有办法做到这一点?

0 个答案:

没有答案