python - 我们如何从配置文件(具有列名)的分隔符的基础上拆分列名

Question

我在 Columns.txt 中有列名如下的文本文件

Employee_ID|X
Employee_Name|X
Age|Y

我正在按照以下方法阅读以上专栏..

with open("Columns.txt") as c:
    columns_list = c.readlines()
columns_list=[col.replace('\n','') for col in columns_list]
print(columns_list)

Output :  ['Employee_ID|X', 'Employee_Name|X', 'Age|Y']

我如何限制输出中具有第二个“|”的列具有“X”的竖线分隔符值。

Example:  Output : should be like [Employee_ID,Employee_Name]

请帮助任何建议或语法

Answer 1

你可以只使用:

with open("Columns.txt") as c:
    columns_list = [col for line in c for col, x in [line.strip().split('|')]
                    if x == 'X']

print(columns_list)

它应该显示:

['Employee_ID', 'Employee_Name']