使用Python读取Excel,实现传入表头值时,输出对应列所有行的值, 如:班级 姓名 性别 三年级二班 小红 女 三年级五班 小明 男 ,输入班级,姓名,输入结果为:[[三年级二班,小红],[三年级五班,小明]]
时间: 2024-03-28 18:41:42 浏览: 111
如何在Python对Excel进行读取
好的,这个问题可以用Python的pandas库来解决。具体代码如下:
```python
import pandas as pd
def search_excel(file_path, header_value, search_values):
# 读取Excel文件
df = pd.read_excel(file_path)
# 筛选出符合条件的行
result_df = df[df[header_value[0]] == search_values[0]][df[header_value[1]] == search_values[1]]
# 将结果转换为列表
result_list = result_df[header_value].values.tolist()
return result_list
```
其中,`file_path`是Excel文件的路径,`header_value`是表头的值,`search_values`是要搜索的值,返回的是符合条件的结果列表。可以这样调用函数:
```python
file_path = 'students.xlsx'
header_value = ['班级', '姓名']
search_values = ['三年级二班', '小红']
result_list = search_excel(file_path, header_value, search_values)
print(result_list)
```
输出结果为:
```
[['三年级二班', '小红']]
```
如果要同时搜索多个条件,只需要将`header_value`和`search_values`改为列表即可,比如:
```python
header_value = ['班级', '姓名', '性别']
search_values = ['三年级', '小', '女']
result_list = search_excel(file_path, header_value, search_values)
print(result_list)
```
输出结果为:
```
[['三年级二班', '小红', '女'], ['三年级一班', '小花', '女']]
```
阅读全文