gpt4 book ai didi

python - 如何删除包含零的 numpy 数组中的一行?

转载 作者:太空狗 更新时间:2023-10-29 20:42:26 25 4
gpt4 key购买 nike

<分区>

我正在尝试编写一个函数来删除所有具有零值的行。这不是来 self 的代码,而是我正在使用的想法的一个示例:

import numpy as np
a=np.array(([7,1,2,8],[4,0,3,2],[5,8,3,6],[4,3,2,0]))
b=[]

for i in range(len(a)):
for j in range (len(a[i])):
if a[i][j]==0:
b.append(i)

print 'b=', b
for zero_row in b:
x=np.delete(a,zero_row, 0)

print 'a=',a

这是我的输出:

b= [1, 3]
a= [[7 1 2 8]
[4 0 3 2]
[5 8 3 6]
[4 3 2 0]]

如何删除 b 中具有索引的行?抱歉,我对此还很陌生,非常感谢任何帮助。

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