item是來自 MySQL 的所有元組y=[1 if (t['color']=='yellow') else -1 for t in item]輸出：[-1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ]像這個，我想做這個for t in item: if(t['color']=='yellow'): y=1 elif(t['color']=='blue'): y=2 else: y=3并將結果放入數組中y=[1 if (t['color']=='yellow') 2 elif (t['color']=='blue') else 3 for t in item]y=np.array(y)print(y)輸出應該是這樣的[ 1 2 3 1 3 2 1 3 1 2 3 2 1 ] 我該怎么做？
2 回答
元芳怎么了
TA貢獻1798條經驗 獲得超7個贊
這是字典的一個很好的用例！
首先，定義一個將顏色名稱映射到數字的字典
colordict?=?{'yellow':?1,?'blue':?2,?'default':?3}然后，使用colordict.get()檢索正確的號碼
y?=?[colordict.get(t['color'],?colordict['default'])?for?t?in?item]
第二個參數是如果在 中找不到.get()則返回的默認值。t['color']colordict
使用字典可以輕松添加更多顏色。您能想象必須編寫一百個嵌套if-elses來支持一百種顏色嗎？！
使用這個虛擬列表進行測試：
item?=?[
????{'color':?'yellow',?'data':?0.5},
????{'color':?'purple',?'data':?0.1},?
????{'color':?'blue',?'data':?0.2},?
????{'color':?'blue',?'data':?0.3},?
????{'color':?'red',?'data':?0.6}
???????]我們得到輸出
[1,?3,?2,?2,?3]
這正是我們所期望的。
如果您defaultdict不想處理.get().
紫衣仙女
TA貢獻1839條經驗 獲得超15個贊
y=[1 if (t['color']=='yellow') else 2 if (t['color']=='blue') else 3 for t in item]
添加回答
舉報
0/150
提交
取消