python - 根据其他行的某些条件获取 pandas 行的内容

Question

我有一个包含以下内容的 pandas DataFrame df1:

Serial N         year         current
   B              10            14
   B              10            16
   B              11            10
   B              11            
   B              11            15
   C              12            11
   C                            9
   C              12            13
   C              12           
   D               3             4

我想计算每个序列唯一序列的出现次数。如果序列号小于 2，我想将该行的年份和当前替换为 nan。我想要这样的东西:

Serial N         year         current
   B              10            14
   B              10            16
   B              11            10
   B              11            
   B              11            15
   C              12            11
   C                             9
   C              12            13
   C              12 
   D              nan           nan      

Answer 1

您可以组合 value_counts、lt 和 reindex 来获取将值更改为 nan 的 bool 数组>，然后使用 loc 进行更改。

serial_filter = df1['Serial N'].value_counts().lt(2).reindex(df1['Serial N'])
df1.loc[serial_filter.values, ['year', 'current']] = np.nan

结果输出:

  Serial N  year  current
0        B  10.0     14.0
1        B  10.0     16.0
2        B  11.0     10.0
3        B  11.0      NaN
4        B  11.0     15.0
5        C  12.0     11.0
6        C   NaN      9.0
7        C  12.0     13.0
8        C  12.0      NaN
9        D   NaN      NaN