ฉันต้องการลบคอลัมน์ที่เลือกใน numpy.array นี่คือสิ่งที่ฉันทำ:
n [397]: a = array([[ NaN, 2., 3., NaN],
.....: [ 1., 2., 3., 9]])
In [398]: print a
[[ NaN 2. 3. NaN]
[ 1. 2. 3. 9.]]
In [399]: z = any(isnan(a), axis=0)
In [400]: print z
[ True False False True]
In [401]: delete(a, z, axis = 1)
Out[401]:
array([[ 3., NaN],
[ 3., 9.]])
ในตัวอย่างนี้ เป้าหมายของฉันคือการลบคอลัมน์ทั้งหมดที่มี NaN's ฉันคาดว่าคำสั่งสุดท้ายจะส่งผลให้:
array([[2., 3.],
[2., 3.]])
ฉันจะทำเช่นนั้นได้อย่างไร?