大家好,我是Python进阶者。
前几天在才哥的Python交流群遇到了一个粉丝提问,提问截图如下:
觉得还挺有意思的,都是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,如下图所示:
这里稍微优化了下,直接得到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)
可以得到如下的结果:
【方法二】代码如下:
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)
后来发现是可以继续优化的,所以就有了上述代码。
大家好,我是皮皮。这篇文章主要盘点了一个Python处理Excel表格数据的问题,文中针对该问题,给出了具体的解析和代码实现,帮助粉丝顺利解决了问题。
最后感谢粉丝提问,感谢【dcpeng】给出的思路和代码解析,感谢【冯诚】等人参与学习交流。