有没有办法计算满足特定条件的列数？

Question

我有以下代码：

import random
import pandas as pd
import numpy as np

def countCol(row):
    count = 0
    for c in range(1, 7):
        if (row['D' + str(c)] < 0):
            count = count + 1
    return(count)
    
data = {'ID': random.sample(range(1, 50), 5),
        'D1': random.sample(range(-5, 5), 5),
        'D2': random.sample(range(-5, 5), 5),
        'D3': random.sample(range(-5, 5), 5),
        'D4': random.sample(range(-5, 5), 5),
        'D5': random.sample(range(-5, 5), 5),
        'D6': random.sample(range(-5, 5), 5)
        }

df = pd.DataFrame(data, columns = ['ID', 'D1', 'D2', 'D3', 'D4', 'D5', 'D6'])

df['Count'] = df.apply(countCol, axis=1)
display(df)

从代码中很明显，我想计算值小于零的列数。 但是，代码使用了for循环和apply function，这让我觉得这是解决这个问题的一种非常低效的方法。

是否有更矢量化或更快的方法？

Answer 1

使用DataFrame.filter和DataFrame.lt来获取正确的列并在我们的条件适合的地方获取布尔值，然后sum以计算列的数量：

df.filter(regex='D\d').lt(0).sum(axis=1)

0    4
1    5
2    1
3    2
4    4
dtype: int64

如果您的ID列始终是第一个，我们可以使用DataFrame.iloc代替：

df.iloc[:, 1:].lt(0).sum(axis=1)

0    4
1    5
2    1
3    2
4    4
dtype: int64

Answer 2

您可以将apply与lambda一起使用，例如

df["Count"] = df.apply(lambda x: sum(x[1:] < 0) , 1)

df
   ID  D1  D2  D3  D4  D5  D6  Count
0  25   0  -3   4  -1   1   4      2
1  45   2   2  -1  -5   3   3      2
2  35   4  -4  -2   0  -4  -3      4
3  41  -3   4   3   4  -1  -1      3
4  34   3  -2   1  -4   2   1      2

Answer 3

如果您想知道满足条件的特定列以及记录数，您可以

    df2=df.apply(lambda s: s.s<0)#Boolean select less than 0
    c=df2.columns.to_numpy()#convert columns to numpy array
    df2['negcount']=[c[i] for i in df2.to_numpy()]#Using list comprehension put columns which meet set conditions in a list
    df2['count']=df2['negcount'].str.len()#Count elements in each list in each row
    df=df.join(df2.iloc[:,-2:])#join a slice of df2  back to df1
    print(df)



   ID  D1  D2  D3  D4  D5  D6              negcount  count
0  24   3   1  -5   0   4   3                  [D3]      1
1  10   1  -3   4  -1  -4   4          [D2, D4, D5]      3
2  30  -1  -1   3   4   2   1              [D1, D2]      2
3  48  -3   0  -3  -3  -2  -1  [D1, D3, D4, D5, D6]      5
4   9   4  -5  -1  -2  -5  -3  [D2, D3, D4, D5, D6]      5