之前要判断有一列是否包含一个字符串,往往写一个for循环,有点麻烦。
今天忽然想到可以用str功能呢。记录一下:
idx_dy = []
for url in df['real_url']:
if "dy/article" in url:
idx_dy.append(1)
else:
idx_dy.append(0)
df['real_url'].str.contains("dy/article")
df['idx_dy'] = df['real_url'].str.contains("dy/article")
df['idx_dy']
df['idx_dy'].value_counts()
实现的功能是一样的,用str.contains更简单,只需要一句。
结果如下