以下示例代码展示了如何使用Python的pand
as
库从CSV文件中提取特定行。CSV文件名为“data.csv”,要提取的行应具有“ID”列值为1和3。首先,将CSV文件读入DataFrame。然后,使用“isin”方法选择具有所需ID列表的行,即ID = 1和3。最后,将所选行写入新的CSV文件“extracted_data.csv”。
import pandas as pd
# 读取CSV文件
data = pd.read_csv("data.csv")
# 选择需要提取的行
selected_rows = data[data["ID"].isin([1, 3])]
# 将选择的行写入新文件
selected_rows.to_csv("extracted_data.csv", index=False)