如何用Python从多个数据集/表格中删除指定列？

原始输出数据集 #

A  B  C
0  2  1  5
1  3  2  3
2  1  3  4
3  4  4  4

   A  B  C
0  2  1  5
1  3  2  3
2  1  3  4
3  4  4  4
4  5  5  5

   A  B  C  D
0  2  1  5  1
1  3  2  3  2
2  1  3  4  3
3  4  4  4  4
4  5  5  5  5

目标输出数据集 #

A  C
0  2  5
1  3  3
2  1  4
3  4  4

   A  C
0  2  5
1  3  3
2  1  4
3  4  4
4  5  5

   A  B  C  D
0  2  1  5  1
1  3  2  3  2
2  1  3  4  3
3  4  4  4  4
4  5  5  5  5

当前使用代码 #

import pandas as pd

df = pd.DataFrame({'A': [2,3,1],
                  'B': [1,2,3],
                  'C': [5,3,4]})
df1 = pd.DataFrame({'A': [4],
                  'B': [4],
                  'C': [4]})
df = df.append(df1)
df = df.reset_index(drop=True)
print(df)

df.loc[df.last_valid_index() + 1] = [5,5,5]
print()
print(df)

df2 = pd.DataFrame({'D': [1,2,3,4,5]})

df = pd.DataFrame.join(df,df2)
print()
print(df)

修改后的代码 #

import pandas as pd

df = pd.DataFrame({'A': [2,3,1],
                  'B': [1,2,3],
                  'C': [5,3,4]})
df1 = pd.DataFrame({'A': [4],
                  'B': [4],
                  'C': [4]})
df = df.append(df1)
df = df.reset_index(drop=True)
# 只输出A、C列，过滤B列
print(df[['A', 'C']])

df.loc[df.last_valid_index() + 1] = [5,5,5]
print()
# 同样只输出A、C列
print(df[['A', 'C']])

df2 = pd.DataFrame({'D': [1,2,3,4,5]})

df = pd.DataFrame.join(df,df2)
print()
# 保留完整列输出
print(df)

修改说明 #

• 前两次print操作时，通过df[['A', 'C']]指定仅选取A、C两列，直接过滤掉B列，匹配目标输出的前两个表格
• 第三次输出join后的DataFrame时，保留所有列（包含B列），与目标的第三个表格要求一致

内容的提问来源于stack exchange，提问作者kefuyuehan