Pandas空/三元组列表转DataFrame存Pickle遇问题求助

嘿，我来帮你搞定这个Pandas的问题！你遇到的大概率是当solutions为空列表时，设置列名会报错的情况，我给你拆解下原因和靠谱的解决办法：

为什么原来的代码会出错？ #

当solutions是空列表时，np.asarray(solutions)会生成一个一维的空数组（比如array([], dtype=float64)），转成DataFrame后，这个表只有索引、没有任何列。这时候你给它赋值3个列名，就会因为「列数不匹配」抛出类似ValueError: Length mismatch的错误。

而且其实你完全不需要先转成numpy数组——Pandas本身就能直接处理由三元组组成的列表，代码可以更简洁！

最优解决方案（一步到位） #

直接在创建DataFrame的时候就指定列名，不管solutions是空还是有数据，都能正常工作：

import pandas as pd

# 直接用列表创建DataFrame，同时指定列名
df = pd.DataFrame(solutions, columns=["Name1", "Name2", "Name3"])
df.to_pickle(path)

• 如果solutions有三元组，会正常生成对应行的DataFrame；
• 如果solutions是空，会生成一个有3列但0行的空DataFrame，后续存pickle也完全没问题。

如果你一定要保留numpy转换的步骤 #

要是你因为某些原因必须先转成numpy数组，可以先判断数组是否为空，再针对性处理：

import numpy as np
import pandas as pd

solutions_arr = np.asarray(solutions)
if solutions_arr.size > 0:
    # 数组非空时，正常创建带列名的DataFrame
    df = pd.DataFrame(solutions_arr, columns=["Name1", "Name2", "Name3"])
else:
    # 数组为空时，直接创建带指定列名的空DataFrame
    df = pd.DataFrame(columns=["Name1", "Name2", "Name3"])
df.to_pickle(path)

小提示 #

后续加载pickle文件时，不管是有数据的表还是空表，都能正常读取到带正确列名的DataFrame，不会影响后续操作~

内容的提问来源于stack exchange，提问作者Mathieu