cannot set a row with mismatched columns

错误背景

在希望将dataframe a 中的特定行移至dataframe b 时出错，记录下自己使用的方法

原错误情况

#dataframe a 已知
a={'ab':[1,2,3],'bb':[3,4,5],'cb':[4,5,6]}
a = pd.DataFrame(a)
print(a)

#dataframe b 是一个空df，希望把a中的特定行放至b
b = pd.DataFrame(columns=a.columns)
#根据网上的方法
size = b.index.size     #获取当前空df的索引值，每找到一个特定行就依次添加，这里只添加一行作为示例
print(a.loc[a['cb']==5]) #查找特定行 没有问题
b.loc[size] = a.loc[a['cb']==5] #出错 cannot set a row with mismatched columns

错误原因

采用a.loc[条件]的方法查找到的特定行是DataFrame的格式，而b.loc[size]，size直接指定了索引，这样查找行的话得到的是df中某一行的值，如：

print(a.loc[a['cb']==5]) 
print(a.loc[1])#依然是上面的例子中的数据

运行结果：

所以

b.loc[size] = a.loc[a['cb']==5]

这个等式两边是不匹配的。因此要解决这个报错，解决方法就是等式右边，查找a中特定行时，采取先获取索引，再查找的方式

解决方法

a={'ab':[1,2,3],'bb':[3,4,5],'cb':[4,5,6]}
a = pd.DataFrame(a)
print(a)
b = pd.DataFrame(columns=a.columns)
size = b.index.size

index = a.loc[a['cb']==5].index.tolist()[0]   #获取a中特定行的索引
print(a.loc[1])
b.loc[size] = a.loc[index]  #两边格式一样
print(b)  #成功！

运行结果：