尝试从python中的列.csv文件打印特定值

Question

I'm trying to get and print all the values (integers) from a column that matches with an specific input. The lenght of the input must be between 3 and 15, as shown below:

import pandas as pd
file = pd.read_csv('filename.csv', encoding="ISO-8859-1", engine='python', sep=';')

def codechoice(): 
 codeNum = int(input("What's the code: "))
 if (len(str(codeNum))) > 3 and (len(str(codeNum))) < 15:
      result = file[file['Column 2'].str.contains(codeNum)]
      print(result)

codechoice()

It's showing the result as expected, but also it's showing all other columns from the csv file of that specific input. Also, if I input, lets say, "100", it is printing all the ocurrencies that has "100" in its length.

The question is: How do I delimit that I want only one specific Column?

Example, if entering the input I just wanted to print the columns 4 and 5? And how to correct the length error?

thank you.

Answer 1

I added some coding to present with some example data so you might need to tweek it a little to match your own data set, but this will return all the rows that match a zip_code that you enter as a df

data = {
    'State' : ['NC', 'SC', 'NC', 'SC'],
    'ZipCode' : [28215, 10101, 28201, 10111]
}
df = pd.DataFrame(data)

def codechoice(): 
    codeNum = int(input("What's the code: "))
    if (len(str(codeNum))) > 3 and (len(str(codeNum))) < 15:
        result = df.loc[df['ZipCode'] == int(codeNum)]
        return result

zip_choice_df = codechoice()
zip_choice_df

From there if you wanted to only limit the return to a specific set of columns you can take another step by adding a df = df[[]] to limit to the specific columns like so:

data = {
    'State' : ['NC', 'SC', 'NC', 'SC'],
    'Country' : ['USA', 'USA', 'USA', 'USA'],
    'ZipCode' : [28215, 10101, 28201, 10111]
}
df = pd.DataFrame(data)

def codechoice(): 
    codeNum = int(input("What's the code: "))
    if (len(str(codeNum))) > 3 and (len(str(codeNum))) < 15:
        result = df.loc[df['ZipCode'] == int(codeNum)]
        return result

zip_choice_df = codechoice()
zip_choice_df = zip_choice_df[['State', 'Country']]
zip_choice_df