python pandas表格格式问题，精简代码

Question

代码如何修改，使房型和销售渠道的数据类型改的和其他一样，把[ ] 去掉，或者有没有更好的方式生成按概率分布的随机数

Answer 1

在函数random_choice中不用列表添加，直接返回符合条件的值即可，或者用 res[0]取出列表元素，在数据框中列数据中就就会去掉[]了。