csv file reading and writing-list, dictionary method



import csv
def csv_write():
    """
    调用writer对象的前提是:需要传入一个文件对象,
    然后才能在这个文件对象的基础上调用csv的写入方法writerow(写入一行)writerrows(写入多行)。
    写入数据的代码如下:

    这段代码我首先定义了写入csv文件的表头、每一列的内容,然后打开一个csv文件,
    将文件对象作为参数传给csv.writer(),最后将表头和每一行的内容写入到csv文件中。

    如果打开csv文件出现空行的情况,那么需要添加一个参数 newline=
    ”(我使用windows出现了这种情况,使用linux mint没有出现)
    :return:
    """

    headers = ['class','name','sex','height','year']

    rows = [
            [1,'xiaoming','male',168,23],
            [1,'xiaohong','female',162,22],
            [2,'xiaozhang','female',163,21],
            [2,'xiaoli','male',158,21]
        ]

    with open('test.csv','w',newline='')as f:
        f_csv = csv.writer(f)
        f_csv.writerow(headers)
        f_csv.writerows(rows)

def dictriter1():
    """
    在写入字典序列类型数据的时候,需要传入两个参数,一个是文件对象——f,
    一个是字段名称——fieldnames,到时候要写入表头的时候,只需要调用writerheader方法,
    写入一行字典系列数据调用writerrow方法,并传入相应字典参数,写入多行调用writerows
    :return:
    """

    headers = ['class', 'name', 'sex', 'height', 'year']

    rows = [
        {
    
    'class': 1, 'name': 'xiaoming', 'sex': 'male', 'height': 168, 'year': 23},
        {
    
    'class': 1, 'name': 'xiaohong', 'sex': 'female', 'height': 162, 'year': 22},
        {
    
    'class': 2, 'name': 'xiaozhang', 'sex': 'female', 'height': 163, 'year': 21},
        {
    
    'class': 2, 'name': 'xiaoli', 'sex': 'male', 'height': 158, 'year': 21},
    ]

    with open('test2.csv', 'w', newline='')as f:
        f_csv = csv.DictWriter(f, headers)
        f_csv.writeheader()
        f_csv.writerows(rows)


def csvreader1():
    """
    读取csv时需要使用reader,并传如一个文件对象,而且reader返回的是一个可迭代的对象,需要使用for循环遍历
    :return:
    """
    with open('test.csv')as f:
        f_csv = csv.reader(f)
        for row in f_csv:
            print(row)

def csvreader2():
    """
    row是一个列表,如果想要查看固定的某列,则需要加上下标,
    例如我想要查看name,那么只需要改为row[1]
    :return:
    """
    with open('test.csv')as f:
        f_csv = csv.reader(f)
        for row in f_csv:
            print(row[1])



if __name__ == '__main__':
    # csv_write()
    # dictriter1()
    # csvreader1()
    csvreader2()

Guess you like

Origin blog.csdn.net/wtt234/article/details/113184567