CODEWITHSUNDEEP
pythoncsvsave
heavyblack1
heavyblack1

Reputation: 82

Save 2D array to CSV

I want function input is list's and each list save as one row.

My attempt to save a two-dimensional array:

my_list = [['\ufeffUser Name', 'First Name', 'Last Name', 'Display Name', 'Job Title', 'Department', 'Office Number', 'Office Phone', 'Mobile Phone', 'Fax', 'Address', 'City', 'State or Province', 'ZIP or Postal Code', 'Country or Region'], ['[email protected]', 'Chris', 'Green', 'Chris Green', 'IT Manager', 'Information Technology', '123451', '123-555-1211', '123-555-6641', '123-555-9821', '1 Microsoft way', 'Redmond', 'Wa',
'98052', 'United States'], ['[email protected]', 'Ben', 'Andrews', 'Ben Andrews', 'IT Manager', 'Information Technology', '123452', '123-555-1212', '123-555-6642', '123-555-9822', '1 Microsoft way', 'Redmond', 'Wa', '98052', 'United States'], ['[email protected]', 'David', 'Longmuir', 'David Longmuir', 'IT Manager', 'Information Technology', '123453', '123-555-1213', '123-555-6643', '123-555-9823', '1 Microsoft way', 'Redmond', 'Wa', '98052', 'United States'], ['[email protected]', 'Cynthia', 'Carey', 'Cynthia Carey', 'IT Manager', 'Information Technology', '123454', '123-555-1214', '123-555-6644', '123-555-9824', '1 Microsoft way', 'Redmond', 'Wa', '98052', 'United States'], ['[email protected]', 'Melissa', 'MacBeth', 'Melissa MacBeth', 'IT Manager', 'Information Technology', '123455', '123-555-1215', '123-555-6645', '123-555-9825', '1 Microsoft way', 'Redmond', 'Wa', '98052', 'United States']]

unifile.dump.excel("file.csv", "array", "utf-8", my_list)

My attempt to save lists to CSV files:

l1 = ["aa", "ftyg"]
l2 = ["fgghg", "ftyfuv"]

unifile.dump.excel("file.csv", "list", "utf-8", l1, l2)

My function

def excel(file_path: str,  mode: str = "array",  t_encoding: str = "utf-8", write_mode: str = "r", *data: list):

    try:
        os.remove(file_path)
    except:
        pass
    if mode == "list":
        with open(file_path, 'w') as csv_file:
            csv_writer = csv.writer(csv_file, delimiter=',')
            for l in data:
                csv_writer.writerow(l)

    elif mode == "array":
        with open(file_path, write_mode, encoding=t_encoding) as csv_file:
            csv_writer = csv.writer(csv_file, delimiter=',')
            for l in data:
                csv_writer.writerow(l)
    else:
        pass

User Name,First Name,Last Name,Display Name,Job Title,Department,Office Number,Office Phone,Mobile Phone,Fax,Address,City,State or Province,ZIP or Postal Code,Country or Region
[email protected],Chris,Green,Chris Green,IT Manager,Information Technology,123451,123-555-1211,123-555-6641,123-555-9821,1 Microsoft way,Redmond,Wa,98052,United States
[email protected],Ben,Andrews,Ben Andrews,IT Manager,Information Technology,123452,123-555-1212,123-555-6642,123-555-9822,1 Microsoft way,Redmond,Wa,98052,United States
[email protected],David,Longmuir,David Longmuir,IT Manager,Information Technology,123453,123-555-1213,123-555-6643,123-555-9823,1 Microsoft way,Redmond,Wa,98052,United States
[email protected],Cynthia,Carey,Cynthia Carey,IT Manager,Information Technology,123454,123-555-1214,123-555-6644,123-555-9824,1 Microsoft way,Redmond,Wa,98052,United States
[email protected],Melissa,MacBeth,Melissa MacBeth,IT Manager,Information Technology,123455,123-555-1215,123-555-6645,123-555-9825,1 Microsoft way,Redmond,Wa,98052,United States

The problem is this code will replace all previous lines with end line.

Upvotes: 0

Views: 88

Answers (1)

Jack Hanson
Jack Hanson

Reputation: 321

The Pandas module is really useful. This worked for me:

import pandas as pd
my_list = pd.DataFrame(my_list)
my_list.to_csv("example.csv")

Upvotes: 1

Related Questions