列表理解迭代 dataframe

Question

我已经编写代码将 dataframe 的一行编码为 json，如下所示：

def encode_df_metadata_row(df):
    return {'name': df['Title'].values[0], 'code': df['Code'].values[0], 'frequency': df['Frequency'].values[0], 'description': df['Subtitle'].values[0], 'source': df['Source'].values[0]}

现在我想通过一些转换将整个 dataframe 编码为 json，所以我写了这个 function：

def encode_metadata_list(df_metadata):
    return [encode_df_metadata_row(df_row) for index, df_row in df_metadata.iterrows()]

然后我尝试使用以下代码调用 function：

df_oodler_metadata = pd.read_csv('DATA\oodler-datasets-metadata.csv')
response = encode_metadata_list(df_oodler_metadata)
print(response)

当我运行此代码时，出现以下错误：

AttributeError: 'str' object has no attribute 'values'

我尝试了很多变体，但我不断收到类似的错误。 有人知道这样做的正确方法吗？

Answer 1

DataFrame.iterrows产生成对的index和row ，其中每一row都是一个系列object。它为每一列存储一个元素，因此encode_df_metadata_row(df) function 中的.values[0]部分变得无关紧要 - 这个的正确形式function 应该是：

def encode_df_metadata_row(row):
    return {'name': row['Title'], 'code': row['Code'], 'frequency': row['Frequency'], 'description': row['Subtitle'], 'source': row['Source']}