如何在Python字典中查找特定键?含CSV转字典实操需求
解决方法
直接在DataFrame中获取列索引(最简便高效):
不需要先转字典,直接利用DataFrame的columns属性自带的get_loc方法,一行代码就能拿到目标列的索引:df = pd.read_csv('data.csv') index = df.columns.get_loc("Wavelength [nm]") print(index)如果一定要用字典的方式实现:
你当前代码里转成的字典,键就是原DataFrame的列名,没必要循环遍历,把键转为列表后直接用index方法查找即可,另外注意别用dict当变量名,会覆盖Python内置的字典类型:df = pd.read_csv('data.csv') data_dict = df.to_dict(orient='dict') columns_list = list(data_dict.keys()) index = columns_list.index("Wavelength [nm]") print(index)
内容的提问来源于stack exchange,提问作者dandan
相关产品推荐
相关产品推荐

