【python】常用文件读取

我就是我 2022-06-09 07:57 151阅读 0赞

## 前言 ##

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

## TXT文件读取 ##

文件内容如下：

![Center][]

代码如下：

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文件读取 ##

文件内容如下：

![Center 1][]

代码如下：

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文件读取 ##

文件内容如下：

![Center 2][]

代码如下：

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文件读取 ##

文件内容如下：

![Center 3][]

代码如下：

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)

[Center]: /images/20220609/0154e0ee215a4930bec3d4c7981e96c9.png
[Center 1]: /images/20220609/e61fd87921054ccd928fcccf43f4cfd7.png
[Center 2]: /images/20220609/5717f695a47c4c188c32a4fc5a6a17b8.png
[Center 3]: /images/20220609/47d828df76764159aa7ab3c58b5886c0.png