用python代码把文件名为“bp_rp_矮星.csv”和“j_k_矮星.csv”的两个csv文件做交集,再把交集的数据另存一个新的文件
时间: 2024-02-16 17:00:40 浏览: 66
Python库 | vip_hci-0.8.1.tar.gz
可以使用Python的pandas库来处理CSV文件,下面是一个实现的示例代码:
``` python
import pandas as pd
# 读取两个CSV文件
bp_rp = pd.read_csv('bp_rp_矮星.csv')
j_k = pd.read_csv('j_k_矮星.csv')
# 将两个DataFrame做交集
intersect = pd.merge(bp_rp, j_k, on='star_id')
# 将交集数据保存为新的CSV文件
intersect.to_csv('intersect.csv', index=False)
```
这段代码将读取名为“bp_rp_矮星.csv”和“j_k_矮星.csv”的两个CSV文件,并将它们的交集保存为名为“intersect.csv”的新文件。其中,我们使用了pandas的merge函数来做DataFrame的交集,on参数指定了用于合并的列名,这里假设两个CSV文件都有一个名为“star_id”的列。最后,我们使用to_csv函数将交集数据保存为新的CSV文件,index=False表示不保存行索引。
阅读全文