使用
np.arange(len(V))用于索引的行数和
V列的：
In [110]: M = [[ 2, 4, 1, 8], .....: [3, 5, 7, 1], .....: [2, 5, 3, 9], .....: [1, 2, 3, 4]]In [111]: V = [2, 1, 0, 1]In [112]:In [112]: M = np.array(M)In [113]: M[np.arange(len(V)),V]Out[113]: array([1, 5, 2, 2])
使用
np.arange(len(V))用于索引的行数和
V列的：
In [110]: M = [[ 2, 4, 1, 8], .....: [3, 5, 7, 1], .....: [2, 5, 3, 9], .....: [1, 2, 3, 4]]In [111]: V = [2, 1, 0, 1]In [112]:In [112]: M = np.array(M)In [113]: M[np.arange(len(V)),V]Out[113]: array([1, 5, 2, 2])
名师互学网 版权所有 (c)2021-2022 ICP备案号：晋ICP备2021003244-6号
