pandas 基础:pandas.DataFrame.apply
表: Triangle
+-------------+------+ | Column Name | Type | +-------------+------+ | x | int | | y | int | | z | int | +-------------+------+ 在 SQL 中,(x, y, z)是该表的主键列。 该表的每一行包含三个线段的长度。
对每三个线段报告它们是否可以形成一个三角形。
以 任意顺序 返回结果表。
思路
三角形任意两边长和大于第三边长和。
pandas.DataFrame.apply
方法签名
DataFrame.apply(func, axis=0, raw=False, result_type=None, args=(), by_row='compat', engine='python', engine_kwargs=None, **kwargs)
关键参数
- func:应用于每一行或列的函数对象。当函数体较短时(单行函数),可以使用 lambda 表达式;
- axis:决定函数应用于行还是列(默认列)。axis 为 0 或 "index" 时,列对象(pandas.Series)将作为参数传递给 func;axis 为 1 或 "columns" 时,行对象(pandas.Series)将作为参数传递给 func。
import pandas as pddef triangle_judgement(triangle: pd.DataFrame) -> pd.DataFrame:triangle['triangle']=triangle.apply(lambda x:"Yes" if x.x+x.y>x.z and x.y+x.z>x.x and x.x+x.z>x.y else 'No',axis='columns')return triangle