比较列值的长度

Question

我有一个有20列的数据框，其中两列是Company1和Company2。我想要一个结果数据帧，其中只有Company1和Company2的长度相差不超过5个字符的行。如何使用熊猫完成此任务？你知道吗

Answer 1

您可以使用.str.len()访问Company列中的字符数，然后使用系列的矢量化减法计算差分，并创建具有子集条件的逻辑向量：

df[abs(df.Company1.str.len() - df.Company2.str.len()) <= 5]