【发布时间】:2019-09-17 18:42:55
【问题描述】:
我有一个 3D 点数组。我想做一些接近 np.histogramdd 所做的事情,但不是每个 bin 中的点数,我想返回其中点的实际坐标:如果没有则为零,如果没有则为精确 xyz只有一个,如果有两个或更多,则为 xyz 平均值,按 bin 指定的顺序。我应该相应地修改 histogramdd 的源代码,还是有更快/更清洁的方法?
【问题讨论】:
我有一个 3D 点数组。我想做一些接近 np.histogramdd 所做的事情,但不是每个 bin 中的点数,我想返回其中点的实际坐标:如果没有则为零,如果没有则为精确 xyz只有一个,如果有两个或更多,则为 xyz 平均值,按 bin 指定的顺序。我应该相应地修改 histogramdd 的源代码,还是有更快/更清洁的方法?
【问题讨论】:
这是一个食谱:
np.digitize获取每个点的每个坐标的bin索引np.ravel_multi_index 转换为平面 bin 索引np.bincount 获取 bin 计数,并使用可选的第二个参数对每个 bin 的坐标求和【讨论】: