2013-03-01 36 views
5

Tôi có một mảng có nhiều mảng và muốn loại bỏ một số cột dựa trên chỉ mục. Có một chức năng trong xây dựng cho nó hoặc một số cách thanh lịch cho một hoạt động như vậy?Python (numpy): thả các cột theo chỉ mục

Cái gì như:

arr = [234, 235, 23, 6, 3, 6, 23] 
elim = [3, 5, 6] 

arr = arr.drop[elim] 

output: [234, 235, 23, 3] 

Trả lời

9

sử dụng numpy.delete, nó sẽ trả về một mảng mới:

import numpy as np 
arr = np.array([234, 235, 23, 6, 3, 6, 23]) 
elim = [3, 5, 6] 
np.delete(arr, elim) 
+0

Cảm ơn! Đây là những gì tôi cần :) – sashkello

Các vấn đề liên quan