Làm cách nào để xóa một hàng trong một mảng có chứa số không?

Question

Tôi đang cố gắng để viết một chức năng để xóa tất cả các hàng, trong đó có một giá trị zero trong Đây không phải là từ mã của tôi, nhưng một ví dụ về ý tưởng Tôi đang sử dụng:.

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 

và đây là đầu ra của tôi:

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

Làm cách nào để loại bỏ các hàng có chỉ mục trong b? Xin lỗi, tôi khá mới với sự trợ giúp này sẽ được thực sự đánh giá cao.

Answer 1

Hình như np.delete thay đổi does't mảng, chỉ cần trả về một mảng mới, vì vậy

Thay vì

x = np.delete(a,zero_row, 0) 

thử

a = np.delete(a,zero_row, 0) 

Answer 2

Tôi nghĩ rằng tôi đã tìm thấy câu trả lời:

dưới dạng @tuxcanfly cho biết tôi đã thay đổi x thành a. Ngoài ra tôi đã loại bỏ vòng lặp for vì nó đã xóa hàng có chỉ số 2 vì một lý do nào đó. Thay vào đó, bây giờ tôi chỉ chọn xóa các hàng bằng cách sử dụng b làm hàm xóa với sử dụng các phần tử trong danh sách để loại bỏ hàng với chỉ mục đó.

mã mới:

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: 
    a=np.delete(a,b, 0) 

print 'a=',a 

và đầu ra:

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

Answer 3

này có thể nhận được một chút quá thông minh, nhưng tôi nghĩ rằng nó giúp dễ đọc (và cho phép bạn để lặp một lần, không hai lần):

#!/usr/bin/env python 

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

for i in a: 
    if not (0 in i): 
     try: 
      b = np.vstack((b, i)) 
     except NameError: 
      b = i 
a = b 
print 'a = ', a 

Trong phiên bản này, bạn lặp qua mỗi hàng và kiểm tra thành viên của 0 trong hàng. Nếu hàng không chứa số không, bạn cố gắng sử dụng np.vstack để nối hàng vào một mảng được gọi là b. Bây giờ, việc khởi tạo b có thể phức tạp hơn, bởi vì nó phải là một mảng numpy với cùng số phần tử như một hàng của một, nhưng thay vì lặp lại hai lần và đi săn tìm giá trị đầu tiên, tôi chỉ dùng thử/trừ để nắm bắt trường hợp nó không tồn tại và chỉ định thay vì sử dụng vstack.

Answer 4

I'm trying to write a function to delete all rows in which have a zero value in.

Bạn không cần phải viết một chức năng cho rằng, nó có thể được thực hiện trong một biểu thức duy nhất:

>>> a[np.all(a != 0, axis=1)] 
array([[7, 1, 2, 8], 
     [5, 8, 3, 6]]) 

đọc như: lựa chọn tuyệt vời a tất cả các hàng đó hoàn toàn khác không.