pandas写入csv

import pandas as pd
write_csv = "train.csv"
"""处理待存数据frame, xmin, xmax, ymin, ymax为list格式"""
column_frame = pd.Series(frame, name="frame")
column_xmin = pd.Series(xmin, name="xmin")
column_xmax = pd.Series(xmax, name="xmax")
column_ymin = pd.Series(ymin, name="ymin")
column_ymax = pd.Series(ymax, name="ymax")
column_classid = pd.Series(class_id, name="class_id")

save = pd.DataFrame({"frame":frame, "xmin":xmin, "xmax":xmax, "ymin":ymin, "ymax":ymax, "class_id":class_id})
save.to_csv(write_csv, index=False, sep=" ")  
"""上面这种方法，列的顺序和设定的不同"""

con = pd.concat([column_frame, column_xmin, column_xmax, column_ymin, column_ymax, column_classid], axis=1)
con.to_csv("train2.csv", index=False, sep=" ")
"""上面这种方法，列的顺序是固定的"""

删除指定包含指定行，参考1，参考2

import pandas as pd

csv_name = "all.csv"
new_name = "deleted_all.csv"
data = pd.read_csv(csv_name)
delete_img = ["F000_00027.jpg", "F000_00069.jpg", "F001_00048.jpg", "F003_00050.jpg",
              "F005_00050.jpg", "F007_00050.jpg", "F011_00050.jpg", "F013_00050.jpg",
              "F015_00050.jpg", "F017_00050.jpg", "F019_00050.jpg", "F027_00050.jpg",
              "F034_00060.jpg", "F036_00060.jpg", "F040_00060.jpg", "F042_00037.jpg",
              "F044_00060.jpg", "F048_00060.jpg", "F050_00037.jpg", "F054_00060.jpg"]
index = data[data["frame"].isin(delete_img)].index   #返回有要删除图像的行号
deleted_data = data.drop(index)
deleted_data.to_csv(new_name, index=False, sep=",")

分别存为train和val，即保留指定行，删除指定行。

new_name = "deleted_all.csv"
deleted_data = pd.read_csv(new_name)
train_name = "train.csv"
val_name = "val.csv"
val_video = ["F055", "F052", "F050", "F040", "F032"]

val_index = deleted_data[deleted_data["frame"].str[:4].isin(val_video)].index
val_data = deleted_data.loc[val_index, :]
train_data = deleted_data.drop(val_index)

train_data.to_csv(train_name, index=False, sep=",")
val_data.to_csv(val_name, index=False, sep=",")