Python creating a list from csv file

Question

I want to import csv file that consist of country names to python. Csv file example:

Afghanistan
Albania
Algeria

I tried using these two pieces of code:

import pandas as pd

df = pd.read_csv('COUNTRIES_LIST.csv')
country_list = df.values.tolist()
print(country_list)

and

import csv

with open('file.csv', newline='') as f:
    reader = csv.reader(f)
    country_list = list(reader)

print(country_list)

Both of these give me a list, but the list has square brackets around each value like this:

[['Afghanistan'], ['Albania'], ['Algeria']]

This doesn't work for me because i'd need to change the values of the list to dictionaries, and for some reason the code that im using only works if there are no brackets in the values.

Is there a way to either import the list without the brackets or delete the brackets from the list? Also, if it's not practical to do that what kind file should i use in place of csv? Thanks!

not_speshal · Accepted Answer · 2022-05-24 17:17:58Z

1

Since your data has just one column, you can squeeze the dataframe to a list before calling tolist:

country_list = df.squeeze().tolist()

answered May 24, 2022 at 17:17

not_speshal

23.2k2 gold badges18 silver badges33 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

ahmedshahriar · Accepted Answer · 2022-05-24 17:13:21Z

0

The reason the first code giving you this list of list output because there is only one column in that CSV file and you are not selecting any specific column

Try this -

import pandas as pd

df = pd.read_csv('COUNTRIES_LIST.csv')
country_list = df.iloc[:, 0].values.tolist()
print(country_list)

answered May 24, 2022 at 17:13

ahmedshahriar

1,0769 silver badges27 bronze badges

Comments

Vakidis · Accepted Answer · 2022-05-24 17:14:17Z

0

#read list of countries from file and store in python list

def readCountries(filename):
    countries = []
    with open(filename) as f:
        for line in f:
            countries.append(line.strip())
    return countries



print(readCountries("COUNTRIES_LIST.csv"))

edited May 24, 2022 at 17:14

answered May 24, 2022 at 17:13

Vakidis

4446 silver badges5 bronze badges

Collectives™ on Stack Overflow

Python creating a list from csv file

3 Answers 3

Comments

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related