多个一维列表(数组)存入csv文件或excel文件

提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档

一维列表(数组)和二维列表(数组)存入csv/excel/txt文件的方法,其实大同小异。只需要稍微修改一下就可以。

定义存储数据的函数,此时data是字典格式的,方便下次直接调用。

def Save_to_Csv(data, file_name, Save_format = 'csv', Save_type = 'col'):
    # data
    # 输入为一个字典,格式: { '列名称': 数据,....}
    # 列名即为CSV中数据对应的列名, 数据为一个列表

    # file_name 存储文件的名字
    # Save_format 为存储类型, 默认csv格式, 可改为 excel
    # Save_type 存储类型 默认按列存储, 否则按行存储

    # 默认存储在当前路径下,可根据自己需要进行更改

    Name = []
    times = 0

    if Save_type == 'col':
        for name, List in data.items():
            Name.append(name)
            if times == 0:
                Data = np.array(List).reshape(-1,1)
            else:
                Data = np.hstack((Data, np.array(List).reshape(-1,1)))

            times += 1

        Pd_data = pd.DataFrame(columns=Name, data=Data)

    else:
        for name, List in data.items():
            Name.append(name)
            if times == 0:
                Data = np.array(List)
            else:
                Data = np.vstack((Data, np.array(List)))

            times += 1

        Pd_data = pd.DataFrame(index=Name, data=Data)

    if Save_format == 'csv':
        Pd_data.to_csv('./'+ file_name +'.csv',encoding='utf-8')
    else:
        Pd_data.to_excel('./'+ file_name +'.xls',encoding='utf-8')

a = [1,2,3]
b = [3,4,5]
c = [6,7,8]
number = {'a':a,'b':b, 'c':c}
Save_to_Csv(data = number, file_name = 'example', Save_format = 'csv',Save_type = 'col')
csvfile=open(r'D:\example.csv',"w",newline="")
writer=csv.writer(csvfile)
writer.writerows(zip(a,b,c))
csvfile.close()

利用pandas库

代码如下(示例):

import pandas as pd
import numpy as np

a = np.array(1,2,3)
b = np.array(3,4,5)
c = np.array(6,7,8)
data  = pd.DataFrame({'a':a, 'b':b,'c':c})
data.to_csv('./example.csv',index=None)
import pandas as pd
import numpy as np

f = pd.read_csv('./example.csv',sep=',',header=None)

d = np.array(2,3,4)
e = np.array(4,5,6)

f['d'] = d
f['e'] = e
f.to_csv('./example.csv', index = False)

注意:当在原有文件的基础上添加列时,不需要使用dataframe,否则会把原始数据覆盖掉。

本文仅仅简单介绍了pandas的使用,而pandas提供了大量能使我们快速便捷地处理数据的函数和方法。

Original: https://blog.csdn.net/weixin_40893164/article/details/125510713
Author: 痴儿CE
Title: 多个一维列表(数组)存入csv文件或excel文件

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/675231/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球