【问题描述】
我想从已有字典中提取出几个键值对构成新的字典,然后为了方便查看,再转成DataFrame格式。
【问题解决】
test_0 = {"id":[1,1,2,3,3,4,5,5],"price":[5,6,8,3,4,6,9,5],"amount":[1,1,2,1,1,1,2,1],"status":['sale','sale','no','no','sale','no','sale','no']}
type(test_0)
dict
我想得到只包含'id','price','amount'的子字典,解决办法如下:
index = {'id','price','amount'} #创建一个set来存储想要的key
test_1 = {key:value for key,value in test_0.items()if key in index} #建立子字典
test_1
{'amount': [1, 1, 2, 1, 1, 1, 2, 1],
'id': [1, 1, 2, 3, 3, 4, 5, 5],
'price': [5, 6, 8, 3, 4, 6, 9, 5]}
出了筛选想要的key之外,还可以筛选value(把if后面的条件改成关于value的即可,如if value > 1)
接着我想把得到的子字典转化为DataFrame格式
pd.DataFrame(test_1)
发现是我们想要的样子。(还好,还好。。。)
问题解决!
(仅供个人学习,不负责任~~~~~~~~~~~~~~~)