gpt4 book ai didi

python - 如何在 numpy 中创建子矩阵

转载 作者:行者123 更新时间:2023-11-28 22:49:08 24 4
gpt4 key购买 nike

我有一个二维 NxM numpy 数组:

a = np.ndarray((N,M), dtype=np.float32)

我想制作一个包含选定数量的列和矩阵的子矩阵。对于每个维度,我都有一个二进制向量或索引向量作为输入。我怎样才能最有效地做到这一点?

例子

a = array([[ 0,  1,  2,  3],
[ 4, 5, 6, 7],
[ 8, 9, 10, 11]])
cols = [True, False, True]
rows = [False, False, True, True]

cols_i = [0,2]
rows_i = [2,3]

result = wanted_function(a, cols, rows) or wanted_function_i(a, cols_i, rows_i)
result = array([[2, 3],
[ 10, 11]])

最佳答案

在numpy中有几种获取子矩阵的方法:

In [35]: ri = [0,2]
...: ci = [2,3]
...: a[np.reshape(ri, (-1, 1)), ci]
Out[35]:
array([[ 2, 3],
[10, 11]])

In [36]: a[np.ix_(ri, ci)]
Out[36]:
array([[ 2, 3],
[10, 11]])

In [37]: s=a[np.ix_(ri, ci)]

In [38]: np.may_share_memory(a, s)
Out[38]: False

请注意,您获得的子矩阵是一个新副本,而不是原始矩阵的 View 。

关于python - 如何在 numpy 中创建子矩阵,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/24241834/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com