CODEWITHSUNDEEP
pythonpandasdataframeconditional-statementsreassign
Jiamei
Jiamei

Reputation: 445

How to reassign value based on condition (Python)

This is a sample dataframe, and I'd like to reassign the values from years b/w 2010-2013 from Company A's A1000 to Company A's B2000. What's a good way to get the result?

enter image description here

Code to generate the dataframe:

df = pd.DataFrame({'Year': [2010, 2010,2010,2010,2010,2011,2011,2011,2012,2012,2012,2012,2012,2013,2013,2013,2014], 
                   'Comapny': ['A','A','A','B','B','A','B','C','A','B','B','B','C','A','B','C','D'],
                   'Code': ['A1000','B2000','C3000','A1000','B2000','B2000','B2000','B2000','A1000','A1000',
                   'B2000','C3000','A1000','B2000','C3000','A1000','A1000'], 
                   'values': [1000,2000,3000,500,1000,2000,4000,4000,1500,4000,2000,6000,1000,5000,2000,1500,2000
]}
                  )

Desired Output:

enter image description here

Upvotes: 1

Views: 48

Answers (1)

Andrej Kesely
Andrej Kesely

Reputation: 195613

You can create a mask and the use df.loc:

mask = ((2010 <= df.Year) & (df.Year <= 2013)) & df.Company.eq('A') & df.Code.eq('A1000')

df.loc[mask, 'Code'] = 'B2000'
print(df)

Prints:

    Year Company   Code  values
0   2010       A  B2000    1000
1   2010       A  B2000    2000
2   2010       A  C3000    3000
3   2010       B  A1000     500
4   2010       B  B2000    1000
5   2011       A  B2000    2000
6   2011       B  B2000    4000
7   2011       C  B2000    4000
8   2012       A  B2000    1500
9   2012       B  A1000    4000
10  2012       B  B2000    2000
11  2012       B  C3000    6000
12  2012       C  A1000    1000
13  2013       A  B2000    5000
14  2013       B  C3000    2000
15  2013       C  A1000    1500
16  2014       D  A1000    2000

Upvotes: 1

Related Questions