将具有选定列标题的CSV读取为Python中的一个CSV文件(按行读取)

问题描述 投票:1回答:1

我有一个问题。我想遍历包含例如名称为“ usr666”,然后仅通过选定的列标题将它们加载到pandas数据帧中,然后将它们合并为一个文件,如以下示例所示:

BT_usr666.csv: 
number|size|person|car    |
---------------------------
31     |2   |Ringo |Tesla  |
82     |3   |Paul  |Audi   |
93     |2   |John  |BMW    |
74     |3   |George|MG     |


RS_usr666.csv:

number|color|person|doors|car    |
---------------------------------
33    |black|Mick  |2    |Porsche|
12    |red  |Keith |4    |Saab   |
55    |blue |Ron   |6    |Volvo  |

into FINAL_usr666.csv

person|car    |
---------------
Ringo |Tesla  |
Paul  |Audi   |
John  |BMW    |
George|MG     |
Mick  |Porsche|
Keith |Saab   |
Ron   |Volvo  |

有什么想法吗?

python pandas
1个回答
0
投票

这可以做到

此将搜索“。”中的文件。即当前目录并查找以usr666开头的文件,然后执行您要求的操作]

import pandas as pd
import os
x=pd.DataFrame()
for filename in sorted(os.listdir(".")):
    if filename.startswith("usr666"):
        y=pd.read_csv(filename)
        selected=y[["person","car"]]
        x=x.append(selected)
        x.to_csv('file1.csv',index=True)
© www.soinside.com 2019 - 2024. All rights reserved.