python读写csv文件实例代码

yipeiwu_com5年前Python基础

Python读取与写入CSV文件需要导入Python自带的CSV模块,然后通过CSV模块中的函数csv.reader()与csv.writer()来进行CSV文件的读取与写入。

写入CSV文件

import csv   # 需要import csv的文件包

out=open("aa.csv",'wb')     # 注意这里如果以‘w'的形式打开,每次写入的数据中间就会多一个空行,所以要用‘wb'

csv_write=csv.write(out,dialect='excel')

# 下面进行具体的内容写入

ss=['a',3]

csv_write.writerow(ss)   #这样ss里面的信息就被写入到aa.csv文件中了,而且是在两列中

                #这里如果需要写入多行,那么就采用循环进行循环输入就可以了

out.close()

读取CSV文件

import csv

out=open("aa.csv",'r')

read_csv=csv.reader(out,dialect='excel')

for line in read_csv:   #循环输出csv中的所有数据

  print line

out.close()

知识点扩展:

import csv

读文件

reader方法

with open('./data.csv', 'r') as f:
 reader = csv.reader(f)
 # python 2
 header = reader.next()
 # python 3x 使用python内置的next
 header = next(reader)
 #获取真正的数据
 for row in reader:
  print(row) # 打印每一行,列表类型,只能按位置索引

稍微介绍一下next(iterable, [default ])函数,它是python3x内置的,用于遍历可迭代对象的方法,其中iterable可以使字符串、列表、字典、元祖、集合等。若default给出,那么遍历到头时返回这个值,不然就抛异常。
这让我想到了字典的get()方法,它也有类似作用。比如,

# 如果dic里面没有key1这个键,那就返回我们指定的0
dic.get('key1', 0)

DictReader方法

写文件

跟读对应, 写操作也有两个方法

writer方法,适用列表的列表

 with open('./data.csv', 'w') as f:
  header = ['col1', 'col2', ...]
  writer = csv.writer(f)
  writer.writeheader(header)
  for row in row_list:
   writer.writerow(row)
  # 或直接调用
  writer.writerows(row_list)
 

DictWriter方法,适用字典列表

 with open('./data.csv', 'w') as f:
  dictwriter = csv.DictWriter(f)
  dictwriter.writeheader(dic_list[0].keys())
  for dic in dic_list:
   writer.writerow(dic)
  # 或者一次性写入,直接调用
   wrier.writerows(dic_list)

都是些比较基础的操作,在这里记录一下。如果你觉得有帮助,就更好了。

相关文章

利用Python求阴影部分的面积实例代码

利用Python求阴影部分的面积实例代码

一、前言说明 今天看到微信群里一道六年级数学题,如下图,求阴影部分面积 看起来似乎并不是很难,可是博主添加各种辅助线,写各种方法都没出来,不得已而改用写Python代码来求面积了 二、...

python实现对求解最长回文子串的动态规划算法

python实现对求解最长回文子串的动态规划算法

基于Python实现对求解最长回文子串的动态规划算法,具体内容如下 1、题目 给定一个字符串 s,找到 s 中最长的回文子串。你可以假设 s 的最大长度为1000。 示例 1: 输入:...

python 函数内部修改外部变量的方法

如果内部修改外部变量需要nonlocal,global def f1(): print("in f1..") num=111 def f2(): nonlocal num...

讲解Python3中NumPy数组寻找特定元素下标的两种方法

讲解Python3中NumPy数组寻找特定元素下标的两种方法

引子 Matlab中有一个函数叫做find,可以很方便地寻找数组内特定元素的下标,即:Find indices and values of nonzero elements。 这个函数...

Python imread、newaxis用法详解

这篇文章主要介绍了python imread、newaxis用法详解,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下 一:imread...