python 读取Dataframe第一列,当i等于第一列某个值,输出Dataframe第二列的值
时间: 2024-09-11 18:08:17 浏览: 84
使用Python向DataFrame中指定位置添加一列或多列的方法
4星 · 用户满意度95%
在Python中,如果你有一个Pandas DataFrame并且想要基于第一列(假设列为A)的特定值(例如'i')获取第二列(假设列为B)的对应值,你可以使用以下代码:
```python
import pandas as pd
# 假设df是你的DataFrame
df = pd.DataFrame({
'A': ['value1', 'i', 'another_value'],
'B': [10, 20, 30]
})
# 定义你想查找的值
target_value = 'i'
# 使用loc或者条件索引来获取对应行的B列值
if target_value in df['A']:
value_in_B = df.loc[df['A'] == target_value, 'B'].values[0]
else:
value_in_B = None # 或者抛出异常,取决于你的需求
print(f"当A列等于{i}时,B列的值是{value_in_B}")
```
这里我们首先检查`target_value`是否在'A'列中存在,如果存在就通过`loc`函数找到该值所在的行,并提取对应的'B'列值。如果`target_value`不在'A'列,我们就处理缺失或错误情况。
阅读全文