python读写数据格式为csv和excel

1. python 存储数据格式为csv和excel

代码：

import os  # 获取当前工作路路径
import pandas as pd
import numpy as np

dataset = input("please input the dataset:")

# 设置文件位置，即当前工作路径下
file_np = os.getcwd()+'\\1_np.csv'
file_pd = os.getcwd()+'\\1_pd.csv'
file_excel = os.getcwd()+'\\'+dataset+'.xlsx'

# 要保留的数据
data_np = np.array([[1,2,3],[4,5,6]])
data_pd = pd.DataFrame(data_np)

# 采用numpy存储csv格式的数据
np.savetxt(fname=file_np,X=data_np,fmt='%.f')
# 采用pandas存储csv格式的数据
data_pd.to_csv(file_pd,index=False) # 数据希尔，index=False表示不加索引
'''产生新的数据，添加到上述文件中已有的数据后面'''
#保存至file文件中，index=False表示文件中不添加索引，header=False表示不添加列名，mode='a+'表示在已有数据基础上添加新数据，并不覆盖已有数据
data2_np = np.array([[7,8,9]])
data2_pd = pd.DataFrame(data2_np)
data2_pd.to_csv(file_pd,index=False,mode='a+',header=False)

'''将数据保存到同一excel不同的sheet中'''

with pd.ExcelWriter(file_excel) as writer:
    data_pd.to_excel(writer,sheet_name='1',float_format='%.2f', header=False, index=False)
    data2_pd.to_excel(writer,sheet_name='2',float_format='%.2f', header=False, index=False)

结果：

2. python读取excel中不同的sheel

def read_excel(path):
    data_xls = pd.ExcelFile(path)
    list1 = data_xls.sheet_names
    data = {}
    arr = np.zeros((2,2,3))
    for i,name in enumerate(list1):
        df = data_xls.parse(sheet_name=name,header=None)
        data[name] = df
        # t = df.values
        arr[i,:,:] = data[name]

    return arr,data

if __name__ == '__main__':
    data = read_excel(file_excel)
    print(data)

3. 参考链接

python中的数据写入(to_csv)、将结果写入excel多个sheet中(to_excel)
Python将数组（矩阵）存成csv文件，将csv文件读取为数组（矩阵）
将numpy数组写入Excel
Python学习——将numpy数组写入Excel