python - 计算csv文件中出现的随机数

Question

我有一个 csv 文件，其中包含以下列:recipe_name(索引)、苹果、柑橘、香蕉、甜瓜，...(以及更多包含水果的列)。对于每个食谱，都有一行显示每种水果的重量百分比。但并非所有食谱都包含所有水果。这些是 csv 文件中的空白区域。它看起来像这样:

recipe_name apple citrus banana melon
pudding 1 0.45 0.78<br/>
pudding 2 0.23 0.15 0.12

现在我想知道每个食谱中的水果数量。所以我想创建一个额外的列，如果第一个食谱包含猕猴桃、香蕉和苹果，那么在这个额外的列中，数字 3 必须在那里。像这样:

recipe_name apple citrus banana melon count
pudding 1 0.45 0.78 2<br/>
pudding 2 0.23 0.15 0.12 3

我试图在 Excel、pandas DataFrame、pandas groupby 中解决这个问题..但我没有找到解决方案..有人有想法吗？

Answer 1

调用count在感兴趣的列上并传递参数 axis=1:

In [217]:
df['count'] = df.ix[:,'apple':].count(axis=1)
df

Out[217]:
  recipe_name  apple  citrus  banana  melon  count
0   pudding 1   0.45    0.78     NaN    NaN      2
1   pudding 2    NaN    0.23    0.15   0.12      3