Python Forum
How to print a column name in csv file - Printable Version

+- Python Forum (https://python-forum.io)
+-- Forum: Python Coding (https://python-forum.io/forum-7.html)
+--- Forum: Data Science (https://python-forum.io/forum-44.html)
+--- Thread: How to print a column name in csv file (/thread-25447.html)



How to print a column name in csv file - Truman - Mar-30-2020

Among other things ( open data sets, explore them, print the first few rows and number of rows and columns ) I should print names of columns
my code looks like this:
def explore_data(dataset, start, end, rows_and_columns=False):
    dataset_slice = dataset[start:end]
    for row in dataset_slice:
        print(row)
        print('\n')  # adds a new (empty) line after each row
    
    if rows_and_columns == True:
        print('Number of rows:', len(dataset))
        print('Number of columns:', len(dataset[0]))
                            
from csv import reader

open_file1 = open('googleplaystore.csv')
read_file1 = reader(open_file1)
android = list(read_file1)
android_header = android[0]
android = android[1:]

open_file2 = open('AppleStore.csv')
read_file2 = reader(open_file2)
apple = list(read_file2)
apple_header = apple[0]
apple = apple[1:]

first = explore_data(android, 1, 3, rows_and_columns=True)
second = explore_data(apple, 1, 3, rows_and_columns=True)
I can't figure out how to print just column names. These two files are regular csv.
I tried smth with
for col_name in row[0]:
    print(col_name)
within a defined function but it gave me uglu result.


RE: How to print a column name in csv file - Larz60+ - Mar-31-2020

with open('googleplaystore.csv') as fp:
    reader = csv.reader(fp):
    header = next(reader)
    print(f"header: {header}")