熊猫没有正确计数行

时间:2019-01-16 14:54:33

标签: python string pandas

所以我有这个数据框:

         filename  width  height    class  xmin  ymin  xmax  ymax
0      128782.JPG    640     512    Panel    36   385   119   510
1      128782.JPG    640     512    Panel   124   388   207   510
2      128782.JPG    640     512    Panel   210   390   294   511
3      128782.JPG    640     512    Panel   294   395   380   510
4      128782.JPG    640     512    Panel   379   398   466   511
5      128782.JPG    640     512    Panel   465   402   553   510
6      128782.JPG    640     512     P+SD   552   402   638   510
7      128782.JPG    640     512     P+SD   558   264   638   404
...
...
57170     128782.JPG    640     512     P+SD    36   242   121   383
57171     128782.JPG    640     512  HS+P+SD    36    97   122   242
57172     128782.JPG    640     512     P+SD   214   106   304   250

在“类别”列中包含的唯一值是“面板”,“ P + SD”和“ HS + P + SD”。我想用这些值来计算多少行,所以我尝试了这个:

print(len(split_df[split_df["class"].str.contains('Panel')]))
print(len(split_df[split_df["class"].str.contains('HS+P+SD')]))
print(len(split_df[split_df["class"].str.contains('P+SD')]))

这给了我这个输出:

56988
0
0

这是不正确的,您可以根据上面提供的DataFrame的片段清楚地看到,为什么对Panel的所有内容都正确计数,而对其他两个“类”名称却不计数?

这是split_df.info的输出:

RangeIndex: 57172 entries, 0 to 57171
Data columns (total 8 columns):
filename    57172 non-null object
width       57172 non-null int64
height      57172 non-null int64
class       57172 non-null object
xmin        57172 non-null int64
ymin        57172 non-null int64
xmax        57172 non-null int64
ymax        57172 non-null int64
dtypes: int64(6), object(2)
memory usage: 3.5+ MB

我无法为自己的生活弄清楚哪里出了问题。任何帮助表示赞赏。

3 个答案:

答案 0 :(得分:6)

pd.Series.str.contains默认具有regex=True。由于+是正则表达式中的特殊字符,因此请使用regex=Falsere.escape\进行转义:

import re
s = pd.Series(['HS+P+SD', 'AB+CD+EF'])

s.str.contains('HS+P+SD').sum()               # 0
s.str.contains('HS+P+SD', regex=False).sum()  # 1
s.str.contains(re.escape('HS+P+SD')).sum()    # 1
s.str.contains('HS\+P\+SD').sum()             # 1
  

我想用这些值计算多少行

如果这是您的核心问题,并且您不希望'P+SD'包括'HS+P+SD',请不要使用str.contains。而是检查是否相等,并对要计算的值使用value_counts

L = ['Panel', 'HS+P+SD', 'P+SD']
counts = df.loc[df['class'].isin(L), 'class'].value_counts()

或者对于所有个计数,只需使用df['class'].value_counts()

答案 1 :(得分:1)

尝试:

print(len(split_df[split_df["class"].str == 'HS+P+SD']))

答案 2 :(得分:1)

使用in进行循环也很简单

int

关于时间安排(如果您想查看此link

sum(['HS+P+SD' in x for x in df['class']])