首页 > 解决方案 > 在python中删除或替换矩阵

问题描述

嗨我有这个矩阵

例如

import numpy as np
a= np.arange(9).reshape(3, 3)

[[1. 2. 3. ]
 [4. 5. 6. ]
 [7. 8. 9. ]]

如何删除所选 col 的最底部值(例如 col = 1),顶部的其余值被向下推,如果有空白,则输入“0”

所以矩阵变成

[[1. 0. 3. ]
 [4. 2. 6. ]
 [7. 5. 9. ]]

标签: pythonnumpy

解决方案


滚动第二列,然后将列的第一个元素设置为零

a = np.arange(1,10).reshape(3, 3)

a[:, 1] = np.roll(a[:, 1], 1)
a[0, 1] = 0

print(a)

输出:

[[1 0 3]
 [4 2 6]
 [7 5 9]]

推荐阅读