python 中 4 列 txt 数据集的一维直方图？

Question

我有一个包含 4 列的文本文件，前 3 列是一个数据点的 x、y 和 z 坐标，第 4 列是该 x、y、z 坐标集处的数据点的值。

例如：

0 1 2 10000
0 1 3 20000
0 2 1 30000
1 0 0 40000
1 1 1 50000

我想制作一个 plot，其横轴为 x 坐标值，纵轴为该位置的 TOTAL 值（x 坐标值）。 这基本上是上面.txt 数据集的 y 和 z 的边缘化直方图。

例如，对于上述数据集，我的 plot 中有 2 个点： (0, 60000)和(1, 90000) ，其中第一个数字代表 x 坐标值，第二个数字代表该值。

我试图阅读有关np.histogramdd function 的信息，但是当我输入 my.txt 数据集时，它会输出一个 4 维张量。 然后，我对其第 2 轴和第 3 轴求和（Matlab 符号）以获得二维张量。 这具有形状(10, 10) 。

我怎样才能从上面获得(0, 60000)和(1, 90000) ？

谢谢你！

Answer 1

其他arrays可以忽略，只做一维直方图。 例如，

In [14]: a = np.array([[0,1,2,10000], [0,1,3,20000], [0,2,1,30000], [1,0,0,40000], [1,1,1,50000]])

In [15]: print(np.histogram(a[:,0], weights=a[:,3]))
(array([60000,     0,     0,     0,     0,     0,     0,     0,     0,
       90000]), array([0. , 0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9, 1. ]))

您可以在np.histogram中使用常用的bins=参数来设置您需要的任何 bin。