Python打开常用文件方法

一、pandas

1.文件读操作

import pandas as pd

#整个文件同时读取
csv_file = pd.read_csv("test.csv")
print(csv_file['data'][0])

#只读取指定列
csv_file = pd.read_csv("test.csv", usecols=['data'])
print(csv_file)

2.文件写操作

import pandas
file = pandas.DataFrame(columns=['index','data'],data=[[0,"今天"],[1,"明天"]])
file.to_csv('test.csv')

二、 open

1.文件读操作

txt_file = open("test.txt",encoding='utf-8')
for line in txt_file.readlines():
    print(line)

2.文件写操作
【注意】此处如果需要写入数字时，需要将数字加引号转为字符型，不然写入会报错。

data = [["0","今天"],["1","明天"]]
with open("text.txt","w",encoding='utf-8') as f:
    for i in data:
        f.writelines(i)

三、xlrd、xlwt

1.文件读操作

import xlrd
xls_file= xlrd.open_workbook('test.xls')
sheet = xls_file.sheet_by_name(xls_file.sheet_names()[0])  #读取表格第一个sheet内容
rows = sheet.nrows   #读取表格的总行数
col_name = sheet.row(0)[1].value   #获取第1列列名
col = []  #用于存储列的所有数据
for i in range(1,rows):   #遍历该列中除列名外所有行数据
    row_values = sheet.row(i)   #获取第i行所有列的数据
    phenomenon = row_values[1]   #取第一列数据
    col.append(phenomenon.value)   #将第一列第i行数据存入col中
print(col_name,'\n',col)

2.文件写操作
【注意】写入表格前，需要保证命名的表格名字事先在存储的当前目录下没有重名的表格存在，否在会报错。

import xlwt
file= xlwt.Workbook(encoding='utf-8')   #定义写入的编码格式为utf-8
sheet = file.add_sheet('sheet1', cell_overwrite_ok=True)   #增加xls文件的sheet
data = [[0,"今天"],[1,"明天"]]   #定义存入数据data
for i in range(len(data)):  #遍历data一维长度
    sheet.write(i+1, 0, data[i][0])  #取data第i个数组中第一个数据，存入表格第i+1行第一列
    sheet.write(i+1, 1, data[i][1]) #取data第i个数组中第二个数据，存入表格第i+1行第二列
col_name=['index','data']     #定义存入数据列名col_name
for i in range(len(col_name)):    #依次存入表格的第一行对应列
	sheet.write(0, i, col_name[i])
file.save('test.xls')   #将写好的表格存在当前目录下