我有一个3D纽比阵列,看起来像这样:
X = [[[10 1] [ 2 10] [-5 3]]
[[-1 10] [ 0 2] [ 3 10]]
[[ 0 3] [10 3] [ 1 2]]
[[ 0 2] [ 0 0] [10 0]]]
首先,我想要X.max(axis=0))沿0轴的最大值:
这给了我:
^{pr2}$下一步是我的问题;我想调用每个10的位置,并从另一个3D数组创建一个新的2D数组,该数组的维数与X相同
例如,具有相同维度的数组如下所示:
Y = [[[11 2] [ 3 11] [-4 100]]
[[ 0 11] [ 100 3] [ 4 11]]
[[ 1 4] [11 100] [ 2 3]]
[[ 100 3] [ 1 1] [11 1]]]
我想找到最大值在X上的位置,并从Y中的数字和位置创建一个2D数组
在这种情况下,答案应该是:
[[11 11] [11 11] [11 11]]
提前感谢您的帮助:)
您可以使用
numpy.argmax
和numpy.indices
来完成此操作。在{a1}为这个问题提供了灵感
你可以试试
相关问题 更多 >
编程相关推荐