python pandas相关问题

Question

假设有这样一个 dataframe，然后其中 id 是不会重复的，是唯一的。
然后另外有一个 id_lst 假设为 [2, 3]
我需要将 id 等于 2, 3的 数据取出来作为一个新的 dataframe1
即新的 datafram1 应该如下图所示:

这需要怎么处理?

Answer 1

# 製造原 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)]

---

我回答過的問題: Python-QA