python 保存数据为excel

python 保存数据为excel

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

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

# 设置文件位置，即当前工作路径下

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

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)