-
Notifications
You must be signed in to change notification settings - Fork 0
/
split.py
51 lines (45 loc) · 1.9 KB
/
split.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
<<<<<<< HEAD
import csv
from collections import defaultdict
# Function to split the CSV by the first column
def split_csv_by_first_column(input_csv):
# Read the input CSV and store data by the first column
with open(input_csv, mode='r', newline='', encoding='utf-8') as file:
reader = csv.reader(file)
header = next(reader)
csv_data = defaultdict(list)
for row in reader:
csv_data[row[0]].append(row)
# Write each group to a separate CSV file
for key, rows in csv_data.items():
output_csv = f'{key}.csv'
with open(output_csv, mode='w', newline='', encoding='utf-8') as file:
writer = csv.writer(file)
writer.writerow(header)
writer.writerows(rows)
print(f'Created file: {output_csv}')
# Replace 'input.csv' with the path to your CSV file
split_csv_by_first_column('rainfall in india 1901-2015.csv')
=======
import csv
from collections import defaultdict
# Function to split the CSV by the first column
def split_csv_by_first_column(input_csv):
# Read the input CSV and store data by the first column
with open(input_csv, mode='r', newline='', encoding='utf-8') as file:
reader = csv.reader(file)
header = next(reader)
csv_data = defaultdict(list)
for row in reader:
csv_data[row[0]].append(row)
# Write each group to a separate CSV file
for key, rows in csv_data.items():
output_csv = f'{key}.csv'
with open(output_csv, mode='w', newline='', encoding='utf-8') as file:
writer = csv.writer(file)
writer.writerow(header)
writer.writerows(rows)
print(f'Created file: {output_csv}')
# Replace 'input.csv' with the path to your CSV file
split_csv_by_first_column('rainfall in india 1901-2015.csv')
>>>>>>> 44fe529aa4d178b868509939f45d54665a8befcb