您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

如何在不使用Python循环的情况下使用每一行中的指定列从矩阵创建向量?

如何在不使用Python循环的情况下使用每一行中的指定列从矩阵创建向量?

使用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])
python 2022/1/1 18:36:28 有431人围观

撰写回答


你尚未登录,登录后可以

和开发者交流问题的细节

关注并接收问题和回答的更新提醒

参与内容的编辑和改进,让解决方法与时俱进

请先登录

推荐问题


联系我
置顶