python - 计算满足两个条件的 Dataframe 列的元素

Question

Barcelona_venues是一个熊猫数据框，包含两列（Neighborhood具有邻域值的列和Venue Category具有场地类别值的列）。两者都包含字符串值。

我想知道 中的每个社区，列表Barcelona_venues中包含的场所数量。interesting_venues

我收到以下错误：

ValueError：Series 的真值不明确。使用 a.empty、a.bool()、a.item()、a.any() 或 a.all()。

---

interesting_venues = ["Café", "Bar", "Market", "Pharmacy", "Train Station", "Supermarket", "Convenience Store", "Flower Shop", "Cafeteria", "Metro Station"]

lst = []

Neigh = Barcelona_venues["Neighborhood"].unique()

for i in Neigh:
    
    count = ((Barcelona_venues[Barcelona_venues["Venue Category"] in interesting_venues]) and (Barcelona_venues[Barcelona_venues["Neighborhood"] == i]))
    lst.append[count]

lst

Answer 1

你可以试试：

interesting_venues = ["Café", "Bar", "Market", "Pharmacy", "Train Station", "Supermarket", "Convenience Store", "Flower Shop", "Cafeteria", "Metro Station"]
lst = []
Neigh = Barcelona_venues["Neighborhood"].unique()

# Loop through each neighborhood
for i in Neigh:
  count = 0
  # Loop through all venue categories for each neighborhood
  for j in Barcelona_venues[Barcelona_venues["Neighborhood"]==i]["Venue Category"]:
    # If it's an interesting category 
    if j in interesting_venues:
      count++
  lst.append[count]