如何在已有Pandas DataFrame中替换列名并修正错位行数据？

问题背景 #

现有一个Pandas DataFrame，初始无列名，且第三行数据左移错位。需要完成两项操作：

• 设置列名为 Process_Date、Transactions、Net_Sales、Third_Party
• 修正第三行的错位，让数据与列对齐

尝试以下代码后触发错误：ValueError: Length mismatch: Expected axis has 1 elements, new values have 4 elements

df.columns = ["Process_Date", "Transactions", "Net_Sales", "Third_Party"]

def shifter(row):
    return np.hstack((delete(np.array(row), [1]), [np.nan]))

mask = df['Transactions'] == '03-Dec'
df.loc[mask, :] = df.loc[mask, :].apply(shift, axis=1)

报错原因 #

1. 列名赋值长度不匹配：初始DataFrame无列名时，若原数据被识别为只有1列（比如读取时未指定header=None，Pandas把第一行当数据、列数为1），直接赋值4个列名就会触发长度不匹配错误。
2. 函数调用错误：代码中apply(shift, axis=1)调用的是Pandas内置的shift函数，而非自定义的shifter，逻辑完全错误。
3. 错位修正逻辑错误：左移错位的行需要右移补NaN在开头，而非删除元素再加NaN在末尾。

正确实现方法 #

步骤1：正确设置列名 #

首先确保读取数据时正确识别列数，再赋值列名：

import pandas as pd
import numpy as np

# 读取无表头的数据，指定header=None让Pandas不把第一行当列名
df = pd.read_csv('your_data_file.csv', header=None)
# 确认原数据列数为4后，设置列名
df.columns = ["Process_Date", "Transactions", "Net_Sales", "Third_Party"]

步骤2：修正错位行 #

左移错位意味着该行数据整体右移一位，开头补NaN。可以通过索引或特定值定位错位行：

方式1：按行索引定位（第三行索引为2，Pandas是0基索引）

wrong_row_idx = 2
# 获取错位行数据并右移修正
corrected_row = np.insert(df.loc[wrong_row_idx].values, 0, np.nan)[:4]
df.loc[wrong_row_idx] = corrected_row

方式2：按特定值定位（比如用户代码中的'03-Dec'）

# 找到错位后跑到Transactions列的目标值
mask = df['Transactions'] == '03-Dec'
# 对匹配行执行右移修正
df.loc[mask] = df.loc[mask].apply(lambda row: np.insert(row.values, 0, np.nan)[:4], axis=1)

完整示例代码 #

import pandas as pd
import numpy as np

# 模拟初始错位数据
data = [
    ['01-Dec', 100, 5000, 200],
    ['02-Dec', 150, 7500, 300],
    ['03-Dec', 120, 6000, np.nan],  # 实际应为[NaN, '03-Dec', 120, 6000]
    ['04-Dec', 180, 9000, 350]
]
df = pd.DataFrame(data, columns=["Process_Date", "Transactions", "Net_Sales", "Third_Party"])

# 修正错位行
mask = df['Transactions'] == '03-Dec'
df.loc[mask] = df.loc[mask].apply(lambda row: np.insert(row.values, 0, np.nan)[:4], axis=1)

print(df)

关键注意事项 #

• 先确认原始DataFrame列数为4，否则设置列名必报错。若列数不对，检查数据读取参数（如分隔符、header设置）。
• 错位修正逻辑要对应：左移错位→右移补NaN在开头；右移错位→左移补NaN在末尾。
• 避免混淆自定义函数与内置函数，调用自定义函数时要写对函数名。

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