鍍金池/ 問(wèn)答/數(shù)據(jù)分析&挖掘  Python/ python pandas相關(guān)問(wèn)題

python pandas相關(guān)問(wèn)題

clipboard.png

假設(shè)有這樣一個(gè) dataframe，然后其中 id 是不會(huì)重復(fù)的，是唯一的。
然后另外有一個(gè) id_lst 假設(shè)為 [2, 3]
我需要將 id 等于 2, 3的 數(shù)據(jù)取出來(lái)作為一個(gè)新的 dataframe1
即新的 datafram1 應(yīng)該如下圖所示:

clipboard.png

這需要怎么處理?

回答
編輯回答
厭惡我
# 製造原 dataframe

from pandas import DataFrame

columns = ['data', 'id', 'key1']
d = {
    'data': [5, 10, 9, 8, 20],
    'id': [1, 2, 3, 4, 5],
    'key1': ['a', 'a', 'b', 'b', 'a']
}

dataframe = DataFrame(d, columns=columns)
# 利用 id_lst 製造新 dataframe

id_lst = [2, 3]
dataframe1 = dataframe[dataframe.id.isin(id_lst)]

我回答過(guò)的問(wèn)題: Python-QA

2017年12月26日 13:24