python 的 pd Dataframe

要如何用A欄位找出特定值

然後修改B欄位的值呢?

例如我想找到 A欄位是 3 與 4的

然後修改這個資料的B欄位為原本資料 X 2

 

import pandas as pd

# 新增 DataFrame
data = {'A': [1, 2, 3, 4, 5],
        'B': [2, 4, 6, 8, 10]}
df = pd.DataFrame(data)

# 找出A欄位是3與4的
tmpDF = df[df['A'].isin([3,4])]

# 將用同樣條件找出來的DF修改後放回去原本的DF
df.loc[df['A'].isin([3,4]), 'B'] = tmpDF['B'] * 2

print(df)

原本我以為修改 df 搜尋出來的tmpDF會影響原本df

結果是不同物件了

所以沒有影響

所以必須要重新指定回去

給大家參考囉