读入火锅店铺数据集(shops_nm.xlsx)为数据框shops_nm。针对菜名列,编写一个函数counter(),可以根据菜名返回菜品数;若菜品数为0,则返回None。 调用函数,查看是否与下面结果一致。
时间: 2024-03-22 13:38:29 浏览: 137
TMS320F28xSerialPeripheralInterface.rar_The Shops
首先,需要使用 `pandas` 库读入 Excel 文件 `shops_nm.xlsx`,并将其转换为数据框 `shops_nm`:
```python
import pandas as pd
shops_nm = pd.read_excel('shops_nm.xlsx')
```
然后,可以编写 `counter()` 函数来计算菜品数:
```python
def counter(dish):
count = shops_nm[shops_nm['菜品名称'].str.contains(dish)]['菜品数量'].sum()
return count if count > 0 else None
```
该函数使用 `pandas` 库中的 `sum()` 函数来计算包含菜名的菜品数量之和,如果菜品数量大于0,则返回该数量,否则返回 `None`。
最后,可以调用该函数来检查结果是否与预期一致,例如:
```python
print(counter('肥牛'))
# 预期输出: 27
print(counter('烤鱼'))
# 预期输出: None
```
如果输出结果与预期一致,则说明函数编写正确。
阅读全文