Pandas --- 取出两列中相同的元素

分别取出两列,将两列转为两个列表先,对两个列表进行比较。

# 看一下都有哪些规格
def cig_data(file_path1, file_path2):
    data = pd.read_excel(file_path1, sheet_name=0, encoding='gbk')
    data.drop_duplicates(subset='物料', keep='first', inplace=True)
    print(data['物料'])
    data_list = data['物料'].values.tolist()      # 将某一列读到列表中
    print(data_list)

    Df = pd.read_excel(file_path2, sheet_name=0, encoding='gbk')
    df_list = Df['物料'].values.tolist()

    a = [x for x in data_list if x in df_list]  # 两个列表表都存在
    print(a)

拓展:

 b = [y for y in (list1 + list2) if y not in a] #两个列表中的不同元素 
 c = [x for x in list1 if x not in list2]  #在list1列表中而不在list2列表中
 d = [y for y in list2 if y not in list1]  #在list2列表中而不在list1列表中

你可能感兴趣的:(程序开发之路)