DataFrame切片和列表的交集

import pandas as pd

# 创建一个示例DataFrame
data = {
    'A': [1, 2, 3, 4],
    'B': [3, 4, 5, 6],
    'C': [4, 5, 6, 7]
}
df = pd.DataFrame(data)

# 定义一个列表
my_list = [3, 4, 7]

# 找到DataFrame列'A'和'B'的值与列表的交集
intersection_AB = df[['A', 'B']].apply(pd.Series.value_counts).index.intersection(my_list).tolist()
print("Intersection of DataFrame columns 'A' and 'B' with the list:", intersection_AB)

# 如果想要找到每一列与列表的交集
intersection_A = df['A'].isin(my_list).tolist()
intersection_B = df['B'].isin(my_list).tolist()
intersection_C = df['C'].isin(my_list).tolist()

print("Intersection of column 'A' with the list:", intersection_A)
print("Intersection of column 'B' with the list:", intersection_B)
print("Intersection of column 'C' with the list:", intersection_C)

# 假设我们有一个包含混合类型的列
df['D'] = ['1', 2, '3', 4]

# 切片后转换数据类型
df['D'] = df['D'].astype(int)

# 重置索引
df_sliced = df.iloc[1:3].reset_index(drop=True)