Python操作csv文件

• 读写操作
• • 法一、使用csv操作
  • • 读取csv文件
    • 将列表写入csv文件
  • 法二、使用自带open函数操作
  • • 读取csv文件
    • 将列表写入csv文件
  • 删除某一行或一列（使用pandas）

import csv

def read_csv(file_path):
    """
    读取csv文件并以列表形式输出
    :param file_path: str
    :return: [list[list]]
    """
    res = []
    csv_file = csv.reader(open(file_path, 'r'))
    for line in csv_file:
        res.append(line)
    return res

def writeData(res,write_path):
    """
    处理数据，并写入新的文件
    :param res: [list[list]]
    :param file_path: str
    :return: bool
    """
    # newline='' 作用是写入后不增加一行空行
    writer=csv.writer(open(write_path,'w',newline=''))
    for line in res:
        writer.writerow(line)

def read_csv(file_path):
    """
    读取csv文件并以列表形式输出
    :param file_path: str
    :return: [list[str]]
    """
    res = []
	with open(file_path, 'r')as f:
	    line="_"
	    while line !="":
	        line = f.readline()
	        res.append(line)
	        print(line,end="")
	return res

def writeData(res,write_path):
    """
    处理数据，并写入新的文件
    :param res: [list[str]]
    :param file_path: str
    :return: bool
    """
    # newline='' 作用是写入后不增加一行空行
    with open(write_path,'w',newline="")as f:
    f.writelines(res)

import pandas as pd
file_path = '../dataset/titanic/w_train.csv'
write_path = '../dataset/titanic/w_test.csv'
# 读取文件数据
data = pd.read_csv(file_path)

data_new = data.drop(["Name"], axis=1)  # 删除表头为Name的这列数据  ,axis=1表示列, axis=0表示删除行，默认值为0
# 或者使用 data.drop([data.columns[index]], axis=1)  # 删除Name这列数据

# 写入文件
data_new.to_csv(write_path, index=0)

将这个使用方法封装了一下

def del_col_row_csv(file_path,write_path=None,index=None,axis=0):
    """
    删除行或者列
    :param file_path:  str 读入文件地址
    :param write_path: str  输出文件地址
    :param index: str/int   删除行或者列的索引（删除列时可以是字符串或者序号，删除行时必须时序号）
    :param axis: int    axis=1 是列， axis=0 是行
    :return:Dataframe   删除后的结果
    """
    if index is None:
        return False
    data = pd.read_csv(file_path)
    if type(index)==str:
        title=index
    else:
        title=data.columns[index]
    data_new = data.drop([title], axis=axis)  # 删除表头为Name的这列数据 ,axis=1表示列, axis=0表示删除行，默认值为0
    if write_path is not None:
        data_new.to_csv(write_path, index=0)
    return data_new