python - 在python pandas中将多个if语句放在一个if语句中
问题描述
我需要将这个大 if 语句从 python 转换为 python pandas。我有一个问题,因为该语句内部有多个 if 语句。
所有值 ( x_value, i_value, u_value, y_val, tb, uvd
) 都是 xlsx 表中的列。我需要使用打印功能中的代码创建一个新列。语句比此处显示的要长
import math
if x_value <= 1.00 and x_value >= 0:
if y_val <= 0.25 and y_val >= 0:
if tb < 0 and i_value == 0:
print('aa1')
elif tb < 0 and uvd > 0:
print('aa2')
elif tb < 0 and uvd < 0:
print('aa3')
elif tb > 0 and uvd < 0:
print('aa4')
elif tb > 0 and uvd > 0:
print('aa5')
elif tb > 0 and u_value > 0:
print('aa6')
else:
print('error')
elif y_val <= 0.50 and y_val >= 0.25:
if tb < 0 and i_value == 0:
print('ab1')
elif tb < 0 and uvd > 0:
print('ab2')
elif tb < 0 and uvd < 0:
print('ab3')
elif tb > 0 and uvd < 0:
print('ab4')
elif tb > 0 and uvd > 0:
print('ab5')
elif tb > 0 and u_value > 0:
print('ab6')
else:
print('error')
elif y_val <= 0.75 and y_val >= 0.50:
if tb < 0 and i_value == 0:
print('ac1')
elif tb < 0 and uvd > 0:
print('ac2')
elif tb < 0 and uvd < 0:
print('ac3')
elif tb > 0 and uvd < 0:
print('ac4')
elif tb > 0 and uvd > 0:
print('ac5')
elif tb > 0 and u_value > 0:
print('ac6')
else:
print('error')
elif y_val <= 1.00 and y_val >= 0.75:
if tb < 0 and i_value == 0:
print('ad1')
elif tb < 0 and uvd > 0:
print('ad2')
elif tb < 0 and uvd < 0:
print('ad3')
elif tb > 0 and uvd < 0:
print('ad4')
elif tb > 0 and uvd > 0:
print('ad5')
elif tb > 0 and u_value > 0:
print('ad6')
else:
print('error')
else:
print('Error!')
elif x_value <= 3.00 and x_value >= 1:
if y_val <= 0.25 and y_val >= 0:
if tb < 0 and i_value == 0:
print('ba1')
elif tb < 0 and uvd > 0:
print('ba2')
elif tb < 0 and uvd < 0:
print('ba3')
elif tb > 0 and uvd < 0:
print('ba4')
elif tb > 0 and uvd > 0:
print('ba5')
elif tb > 0 and u_value > 0:
print('ba6')
else:
print('error')
elif y_val <= 0.50 and y_val >= 0.25:
if tb < 0 and i_value == 0:
print('bb1')
elif tb < 0 and uvd > 0:
print('bb2')
elif tb < 0 and uvd < 0:
print('bb3')
elif tb > 0 and uvd < 0:
print('bb4')
elif tb > 0 and uvd > 0:
print('bb5')
elif tb > 0 and u_value > 0:
print('bb6')
else:
print('error')
elif y_val <= 0.75 and y_val >= 0.50:
if tb < 0 and i_value == 0:
print('bc1')
elif tb < 0 and uvd > 0:
print('bc2')
elif tb < 0 and uvd < 0:
print('bc3')
elif tb > 0 and uvd < 0:
print('bc4')
elif tb > 0 and uvd > 0:
print('bc5')
elif tb > 0 and u_value > 0:
print('bc6')
else:
print('error')
elif y_val <= 1.00 and y_val >= 0.75:
if tb < 0 and i_value == 0:
print('bd1')
elif tb < 0 and uvd > 0:
print('bd2')
elif tb < 0 and uvd < 0:
print('bd3')
elif tb > 0 and uvd < 0:
print('bd4')
elif tb > 0 and uvd > 0:
print('bd5')
elif tb > 0 and u_value > 0:
print('bd6')
else:
print('error')
else:
print('Error!')
else:
print('ERROR')
有什么建议么?
解决方案
- 我想你可能想在熊猫中获得一些功能。
-
- 生成玩具数据集
import numpy as np
import pandas as pd
data = pd.DataFrame(np.random.random(size=(10, 6)))
data.columns = ['x_value', 'i_value', 'u_value', 'y_val', 'tb', 'uvd']
data
-
- 生成一个小函数
def cal_value(x):
if x['x_value'] < 0.5 and x['y_val'] > 0.5:
return 'type1'
elif x['i_value'] < 0.5 and x['tb'] > 0.5:
return 'type2'
else:
return 'type3'
-
apply
在 DataFrame 中使用函数
data['type'] = data.apply(lambda x: cal_value(x), axis=1)
data
如果想了解更多详细信息apply
,请单击熊猫文档链接:https ://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.DataFrame.apply.html
推荐阅读
- docker - Kubernetes 服务不会在 Pod 之间分发请求
- amazon-web-services - Cloudwatch 访问超过 5 次的用户数
- java - 显示重复项目的回收站视图
- sql - 使用 PostgreSQL 从数组中获取标签
- r - 下采样 R 时训练/测试样本不是随机的
- python - 为什么我的代码在 .py 文件中运行时可以工作,但在 Python 解释器中返回 SyntaxError?
- cql - cqlsh:错误:没有这样的选项:--request-timeout
- python - SyntaxError (Python Jupyter Notebook)
- regex - 在 VS Code 中查找缺少两个右括号中的一个的字段
- javascript - 从 JSON 数组中删除元素