gpt4 book ai didi

python - 在 Python 中从矩阵中选择列向量

转载 作者:太空宇宙 更新时间:2023-11-03 14:09:38 24 4
gpt4 key购买 nike

我想在 Python/numpy 中索引矩阵中的列向量,并将其作为列向量而不是一维数组返回。

x = np.array([[1,2],[3,4]])
x[:,1]
>array([2, 4])

给予

np.transpose(x[:,1])

不是解决方案。按照 numpy.transpose 文档,这将返回一个行向量(一维数组)。

最佳答案

几个选项-

x[:,[1]]
x[:,None,1]
x[:,1,None]
x[:,1][:,None]
x[:,1].reshape(-1,1)
x[None,:,1].T

关于python - 在 Python 中从矩阵中选择列向量,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/39914049/

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