【python】常用文件读取

前言

本篇文章介绍通过python读写各种文件，涉及TXT，Excel，CSV，JSON文件。

TXT文件读取

文件内容如下：

代码如下：

import numpy as np

# Method 1
file1 = open("TXT.txt")
while True:
	line = file1.readline()
	print(line)
	if not line:
		break
file1.close()

# Method 2
for line in open("TXT.txt"):
	print(line)

# Method 3
with open("TXT.txt",'r') as f:
	data = f.read()
	print(data)

'''
读取文件的3种方法：
	read()将文本中所有行读到一个字符串中去
	readline()一行一行读，在读行过程中可以跳过特定行
	readlines()将文本中所有行读到一个list中，文本文件每一行是list的一个元素
'''
# Read 
a = np.loadtxt('TXT.txt')
b = a.reshape(3,3)
c = a.reshape(-1,1,3)
print(b,c)

with open('TXT.txt','r') as f:
	data = f.readlines()
	for line in data:
		item = line.split()
		#item = map(float, item)
		print(float(item[1])+1)

Excel文件读取

文件内容如下：

代码如下：

import os       
import xlrd

data = xlrd.open_workbook('EXCEL.xlsx')
data = data.sheet_by_index(0)
nrows=data.nrows
ncols=data.ncols
print(nrows,ncols) 
col_score = data.col_values(1)
row_values = data.row_values(0)
print(col_score,row_values)

data=xlrd.open_workbook('EXCEL.xlsx')     
table=data.sheets()[0] 
data_list=[]    
for i in range(4):
	data_list.extend(table.row_values(i))
print(data_list)

CSV文件读取

文件内容如下：

代码如下：

import csv

csvFile = open("CSV.csv", "r")
reader = csv.reader(csvFile)
data = []
for item in reader:
	if reader.line_num == 1:
		continue
	data.extend(item)
csvFile.close()
print(data)

csvFile = open("CSV.csv", "r")
reader = csv.reader(csvFile)
# 建立空字典
result = {}
for item in reader:
    if reader.line_num == 1:
        continue
    result[item[1]] = item[2]

csvFile.close()
print(result)

JSON文件读取

文件内容如下：

代码如下：

import json
JSON = {
    'name' : 'ACME',
    'shares' : 100,
    'price' : 542.23
}

json_str = json.dumps(JSON)

data2 = json.loads(json_str)

# Writing JSON data
with open('JSON.json', 'w') as f:
    a = json.dump(JSON, f)

# Reading data back
with open('JSON.json', 'r') as f:
    data = json.load(f)
print(data)