如何移除Python列表元素且不返回None?(DataFrame索引场景)
问题解决方法
为什么返回None?
列表的remove()方法是原地修改原列表,它的返回值固定为None,不会返回修改后的列表,所以直接打印这个方法的调用结果自然是None。
正确获取移除元素后的列表的两种方法:
- 先将索引转为列表并保存,再执行移除操作,最后打印保存的列表:
# 先把索引转成列表存起来 index_list = df.index.to_list() # 原地移除'SalePrice' index_list.remove('SalePrice') # 打印修改后的列表 print(index_list)
- 使用列表推导式生成新列表(这种方式不会修改原列表,适合需要保留原索引列表的场景):
# 直接生成排除'SalePrice'的新列表 index_list = [col for col in df.index.to_list() if col != 'SalePrice'] print(index_list)
内容的提问来源于stack exchange,提问作者Katsu
相关产品推荐
相关产品推荐

