How to append an empty column in a dataframe in Python Pandas

Question

I have a DataFrame in Python. It has couple of columns, name 'Item','FRUIT', 'Days before it get spoiled', 'TEMP' etc.

I want to write a program which generates a 'CODE'. for each row in the DataFrame. So, I generated an empty column Named 'CODE'.

Item	FRUIT	DAYs	TEMP
Item_1	Apple	5	20
Item_2	Banana	15	35

Now I have couple of coditions, to generate a code and fill the empty column called 'CODE' using append() method. Following is the program I came up with. But it is not appending in the 'CODE' column. I am new in Programing and I need it for reasearch. Any help for program is more than welcome

from csv import reader

file = open('customerprocessrequirement1.csv', 'r')
Reader = csv.reader(file)
df = pd.read_csv('customerprocessrequirement1.csv')
df['CODE'] = ''
ndf = df
AMGT_code = []

for row in Reader:
    if row[1] == 'APPLE':
        CODE.append(1)
    else:
        CODE.append(2)
    if row[2] <= '10':
        CODE.append(0)
    else row[2] > '10':
        AMGT_code.append(1)
    if row[3] <= '20':
        CODE.append(1)
    else:
        CODE.append(2)

Pls state your rules clearly.

ironv
– ironv

2022-12-07 18:58:14 +00:00
Commented Dec 7, 2022 at 18:58 — ironv
– ironv, Commented Dec 7, 2022 at 18:58

ironv · Accepted Answer · 2022-12-07 19:03:56Z

1

Here's some sample code which you can tweak after the rules have been clearly articulated.

df = pd.DataFrame([["Item_1","Apple",5,20],["Item_2","Banana",15,35]], 
                  columns=["Item","Fruit","Days","Temp"])
df["Code"] = 0

df.loc[df["Fruit"]=="Apple", "Code"] = 1
df.loc[df["Temp"]>20, "Code"] = 2
df.loc[(df["Fruit"]!="Apple") & (df["Days"]>10), "Code"] = 3
print(df)

answered Dec 7, 2022 at 19:03

ironv

1,07812 silver badges28 bronze badges

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

1 Comment

B M Over a year ago

CODE: APPLE =1; Banana = 2; IF temperature < 5

Collectives™ on Stack Overflow

How to append an empty column in a dataframe in Python Pandas

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related