天天看点

# yyds干货盘点 # 盘点一个Python处理Excel两列单元格中有类似字符串就返回1,没有就返回0的操作

大家好,我是Python进阶者。

一、前言

前几天在才哥的Python交流群遇到了一个粉丝提问,提问截图如下:

# yyds干货盘点 # 盘点一个Python处理Excel两列单元格中有类似字符串就返回1,没有就返回0的操作

觉得还挺有意思的,都是​

​Pandas​

​基础操作,这里拿出来给大家一起分享下。

二、实现过程

这里【dcpeng】给了一个代码,如下所示:

import pandas as pd
df = pd.read_excel('test.xlsx')
df["标记列"] = df[["字符串1", "字符串2"]].apply(lambda x: len(set(x['字符串1']) & set(x['字符串2'])) > 0, axis=1)
print(df)      

不过得到的是​

​True​

​和​

​False​

​,如下图所示:

# yyds干货盘点 # 盘点一个Python处理Excel两列单元格中有类似字符串就返回1,没有就返回0的操作

这里稍微优化了下,直接得到0,1,三个方法,一起学习下。

【方法一】代码如下:

import pandas as pd

df = pd.read_excel('test.xlsx')
df["标记列"] = df[["字符串1", "字符串2"]].apply(lambda x: len(set(x['字符串1']) & set(x['字符串2'])) > 0, axis=1)
bool_map = {True: 1, False: 0}
df['new_标记列'] = df['标记列'].map(bool_map)
print(df)      

可以得到如下的结果:

# yyds干货盘点 # 盘点一个Python处理Excel两列单元格中有类似字符串就返回1,没有就返回0的操作

【方法二】代码如下:

import pandas as pd

df = pd.read_excel('test.xlsx')
df["标记列"] = df[["字符串1", "字符串2"]].apply(lambda x: 1 if len(set(x['字符串1']) & set(x['字符串2'])) > 0 else 0, axis=1)      

同样可以得到相同的结果。

【方法三】代码如下:

import pandas as pd

df = pd.read_excel('test.xlsx')
df["标记列"] = df.apply(lambda x: 1 if len(set(x['字符串1']) & set(x['字符串2'])) > 0 else 0, axis=1)
print(df)      

后来发现是可以继续优化的,所以就有了上述代码。

三、总结