发出调用函数的问题

Question

我有一个名为ro的数据框，其中包含汽车零件的所有声明，我现在想要的是创建一个名为part_dataframe的函数，我可以将原始ro投影到一个新的仅包含特定部分的数据框，假设压缩器的子集名称为comp_claims

我的功能是：

def part_dataframe(first_frame, subset, type_number, number):
   subset = first_frame.loc[first_frame[type_number] == number]
   subset = subset.reset_index(drop=True)
   subset['word'] = subset.Comment.str.split().apply(lambda x: pd.value_counts(x).to_dict())

当我试图调用该函数时：

part_dataframe(ro, comp_claims, 'Part No.', '97701')

我收到以下错误：

---------------------------------------------------------------------------
NameError                                 Traceback (most recent call last)
<ipython-input-17-65cf8428af26> in <module>()
----> 1 part_dataframe(ro, comp_claims, 'Part No.', '97701')

NameError: name 'comp_claims' is not defined

我该如何解决？ 提前谢谢

Answer 1

ro = pd.DataFrame(
    {'Part No.': np.arange(10)}
)

def part_dataframe(first_frame, type_number, number):
    return first_frame.loc[first_frame[type_number] == number]

subset = part_dataframe(ro, 'Part No.', 3)
subset