如何比较数据框中的两列，检查它们之前是不是存在？

Posted 2023-03-12

【中文标题】如何比较数据框中的两列，检查它们之前是不是存在？

【英文标题】：How do i compare two columns in a dataframe, checking if they exist before?如何比较数据框中的两列，检查它们之前是否存在？

【发布时间】：2021-09-06 00:50:42

【问题描述】：

我有一个具有以下列级别名称的 5 级 MultiIndex 数据框：

names=['Point', 'Method', 'Site', 'Direction', 'Damping']

和索引名称frequency。

每一列都是一个光谱，我的目标是比较每个“点”的不同光谱。 更具体地说，我想比较每一点：

df.loc[:,(Specific Point, 1, Big site, Horizontal, 0.05)

与

df.loc[:,(Specific Point, 2, Small Site, Horizontal, 0.05)

因为这个数据框是合并的结果，所以列并不总是存在的，我需要根据不同的条件比较额外的点。

我的目标是创建一个具有不同列的新数据框，指定是否：频谱 B 在某个点超过频谱 A，如果频谱 B 最大值大于频谱 A 最大值等。

问题是我不知道如何检查我需要比较的列是否存在。如果不是，则所需的操作是通过该点并开始下一个。

我尝试使用点名称创建一个循环，然后调用特定列，但由于有时它们不存在，我得到一个 KeyError 指示标签不存在。

【问题讨论】：

您可以使用if col in df.columns: ... 之类的东西，其中col 是您要检查的列的标签？

感谢您的回答，我已经尝试了您所写的内容，我遇到的问题是，如果条件不存在时，此 if 条件下的 'col' 调用可能会失败，从而导致 KeyError 并采取我们回到原来的问题。

【参考方案1】：

我通过使用 try 成功解决了我的问题，但在我的循环中除外，例如：

try:
    # trying to call both columns needed, if fail then go to except.
    SpectrumA = df.loc[:,(Specific Point, 1, Big site, Horizontal, 0.05)
    SpectrumB = df.loc[:,(Specific Point, 2, Small Site, Horizontal, 0.05)
    # comparing both columns in case of success, and writing the result in a
    # new dataframe.
    spectrum_compare(newdf, i, SpectrumA, SpectrumB)

except KeyError as error:
    print(error)