构建一个函数,输入dataframe A 和 列名的列表, 输出A中列表中的列名都不为空, 但其他列为空的子表。
时间: 2024-09-30 15:11:32 浏览: 29
pandas 空的dataframe 插入列名的示例
你可以使用pandas库来构建这样一个函数,下面是一个简单的Python示例:
```python
import pandas as pd
def filter_empty_columns(df, column_list):
# 检查DataFrame df中给定列名是否都在df中存在
columns_in_df = set(df.columns).intersection(column_list)
# 创建一个新的DataFrame,只包含非空列
filtered_df = df.loc[:, columns_in_df]
return filtered_df
# 使用示例
# 假设dataframe A
A = pd.DataFrame({
'Column1': [1, 2, None],
'Column2': [None, 4, 5],
'Column3': ['a', 'b', 'c'],
'EmptyColumn': [],
})
column_list = ['Column1', 'Column2', 'EmptyColumn']
filtered_table = filter_empty_columns(A, column_list)
print(filtered_table)
#
阅读全文