列出理解以及如何将其与 pandas 一起使用 [关闭]

Question

我仍然很难理解列表推导。

以下面的列表为例：

lst = ["Medical Center", "Playground", "Park", "Soccer Field", "Swim School", "Summer Camp", "Museum"]

Qn 1) 如何使用列表推导来遍历上述列表并检查具有一百个元素的熊猫数据框（比如说）？

Qn 2) 我想知道是否有一个很好的资源可以更清晰地理解列表；我看过一些网上的资源，虽然简单的操作更容易理解，但一旦变成列表中的列表（列表中，列表中等），那么它就像泥巴一样清晰

Answer 1

您可以这样做，例如，如果您的数据框中只有以字母“P”开头的元素，并使用它通过列表理解过滤您的列表：

import pandas as pd

lst = ["Medical Center",
       "Playground",
       "Park",
       "Soccer Field",
       "Swim School",
       "Summer Camp",
       "Museum"
       ]

# filtered_lst = [elem for elem in lst if elem.startswith('P')]

df = pd.DataFrame(["Playground", "Park"])

# Extract dataframe values into a flattened list:
df_values = [k for i in df.values for k in i]

# Compare original list elements to dataframe values:
common_elements = [elem for elem in lst if elem in df_values]

print(common_elements)

将导致：

['Playground', 'Park']