import numpy as np
W = [
[1,2,3,4],
[1,2,3,4],
[1,2,3,4],
]
W1 = np.array(W) # 切记要把矩阵转换为numpy形式,否则无法输出
u =[1,2]
print(W1[:,u])
if __name__ == '__main__':
pass
python-输出自定义矩阵某行、某列
最新推荐文章于 2022-10-30 01:53:53 发布
import numpy as np
W = [
[1,2,3,4],
[1,2,3,4],
[1,2,3,4],
]
W1 = np.array(W) # 切记要把矩阵转换为numpy形式,否则无法输出
u =[1,2]
print(W1[:,u])
if __name__ == '__main__':
pass