ค้นหาดัชนีจำนวนเต็มของแถวด้วย NaN ในแพนด้าดาต้าเฟรม


96

ฉันมี DataFrame แพนด้าแบบนี้:

                    a         b
2011-01-01 00:00:00 1.883381  -0.416629
2011-01-01 01:00:00 0.149948  -1.782170
2011-01-01 02:00:00 -0.407604 0.314168
2011-01-01 03:00:00 1.452354  NaN
2011-01-01 04:00:00 -1.224869 -0.947457
2011-01-01 05:00:00 0.498326  0.070416
2011-01-01 06:00:00 0.401665  NaN
2011-01-01 07:00:00 -0.019766 0.533641
2011-01-01 08:00:00 -1.101303 -1.408561
2011-01-01 09:00:00 1.671795  -0.764629

มีวิธีที่มีประสิทธิภาพในการค้นหาดัชนี "จำนวนเต็ม" ของแถวด้วย NaN หรือไม่? [3, 6]ในกรณีนี้ผลลัพธ์ที่ต้องการควรจะเป็น


12
หากคุณต้องการเลือกแถวด้วย nan คุณสามารถทำได้df[np.isnan(df['b'])]
lazy1

4
ติดตามจาก @ lazy1 - แทนการใช้numpy's isnanคุณยังสามารถใช้df['b'].isnull()
jmetz

คำตอบ:


48

สำหรับ DataFrame df:

import numpy as np
index = df['b'].index[df['b'].apply(np.isnan)]

จะให้ข้อมูลMultiIndexที่คุณสามารถใช้ในการจัดทำดัชนีกลับเข้าไปdfเช่น:

df['a'].ix[index[0]]
>>> 1.452354

สำหรับดัชนีจำนวนเต็ม:

df_index = df.index.values.tolist()
[df_index.index(i) for i in index]
>>> [3, 6]

1
ในฐานะที่ใช้งานง่ายเป็นixเสียงด้วยเหตุผลบางอย่างที่ดูเหมือนจะได้รับการเลิกในความโปรดปรานของiloc
กระวาน

145

นี่คือวิธีแก้ปัญหาที่ง่ายกว่า:

inds = pd.isnull(df).any(1).nonzero()[0]

In [9]: df
Out[9]: 
          0         1
0  0.450319  0.062595
1 -0.673058  0.156073
2 -0.871179 -0.118575
3  0.594188       NaN
4 -1.017903 -0.484744
5  0.860375  0.239265
6 -0.640070       NaN
7 -0.535802  1.632932
8  0.876523 -0.153634
9 -0.686914  0.131185

In [10]: pd.isnull(df).any(1).nonzero()[0]
Out[10]: array([3, 6])

29
ฉันลงเอยด้วยการใช้สิ่งนี้:np.where(df['b'].notnull())[0]

ขอบคุณ.nonzero()[0]ดีกว่า[i for i, k in enumerate(mask) if k])
Winand

2
คุณอาจทำให้สิ่งนี้ง่ายขึ้น:r, _ = np.where(df.isna())
cs95

2
เพิ่ม.to_numpy()เพื่อแปลงในอาร์เรย์ numpy ก่อน -pd.isnull(df).any(1).to_numpy().nonzero()
7bStan

14

โซลูชันบรรทัดเดียว อย่างไรก็ตามใช้ได้กับคอลัมน์เดียวเท่านั้น

df.loc[pandas.isna(df["b"]), :].index

นี่คือสิ่งที่ฉันกำลังมองหา ฉันทำให้มันกลายเป็นรายการโดยการห่อไว้ในlist(...)ลักษณะนี้:list(df.loc[pandas.isna(df["b"]), :].index)
Daniel Butler

10

และในกรณีถ้าคุณต้องการหาพิกัดของ 'nan' สำหรับคอลัมน์ทั้งหมดแทน (สมมติว่าเป็นตัวเลขทั้งหมด) ไปที่นี่:

df = pd.DataFrame([[0,1,3,4,np.nan,2],[3,5,6,np.nan,3,3]])

df
   0  1  2    3    4  5
0  0  1  3  4.0  NaN  2
1  3  5  6  NaN  3.0  3

np.where(np.asanyarray(np.isnan(df)))
(array([0, 1]), array([4, 3]))

9

ไม่ทราบว่าสายเกินไปหรือไม่ แต่คุณสามารถใช้ np.where เพื่อค้นหาดัชนีที่ไม่ใช่ค่าดังต่อไปนี้:

indices = list(np.where(df['b'].isna()[0]))

4

ในกรณีที่คุณมีดัชนีวันที่และเวลาและคุณต้องการมีค่า:

df.loc[pd.isnull(df).any(1), :].index.values

4

นี่คือการทดสอบสำหรับบางวิธี:

%timeit np.where(np.isnan(df['b']))[0]
%timeit pd.isnull(df['b']).nonzero()[0]
%timeit np.where(df['b'].isna())[0]
%timeit df.loc[pd.isna(df['b']), :].index

และการกำหนดเวลาที่สอดคล้องกัน:

333 µs ± 9.95 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)
280 µs ± 220 ns per loop (mean ± std. dev. of 7 runs, 1000 loops each)
313 µs ± 128 ns per loop (mean ± std. dev. of 7 runs, 1000 loops each)
6.84 ms ± 1.59 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)

ก็จะปรากฏว่า pd.isnull(df['DRGWeight']).nonzero()[0]จะชนะทั้งวันในแง่ของเวลา แต่วิธีใด ๆ ในสามอันดับแรกมีประสิทธิภาพเทียบเท่ากัน




1

ฉันกำลังมองหาดัชนีของแถวทั้งหมดที่มีค่า NaN
โซลูชันการทำงานของฉัน:

def get_nan_indexes(data_frame):
    indexes = []
    print(data_frame)
    for column in data_frame:
        index = data_frame[column].index[data_frame[column].apply(np.isnan)]
        if len(index):
            indexes.append(index[0])
    df_index = data_frame.index.values.tolist()
    return [df_index.index(i) for i in set(indexes)]

0

ให้ dataframe จะชื่อDFและคอลัมน์ที่น่าสนใจ (เช่นคอลัมน์ที่เรากำลังพยายามที่จะหา nulls ) เป็น'B' จากนั้นตัวอย่างต่อไปนี้จะให้ดัชนีที่ต้องการของ null ในดาต้าเฟรม:

   for i in range(df.shape[0]):
       if df['b'].isnull().iloc[i]:
           print(i)
โดยการใช้ไซต์ของเรา หมายความว่าคุณได้อ่านและทำความเข้าใจนโยบายคุกกี้และนโยบายความเป็นส่วนตัวของเราแล้ว
Licensed under cc by-sa 3.0 with attribution required.