dataframe,如果列A取值的后面2个字符是“06”或者“09”,那么就删掉该行
时间: 2023-10-13 18:06:20 浏览: 25
可以使用pandas库中的str属性和drop方法实现该需求,代码如下:
```python
import pandas as pd
# 创建示例数据
df = pd.DataFrame({'A': ['abc06', 'def09', 'ghi07', 'jkl10'], 'B': [1, 2, 3, 4]})
# 删除符合条件的行
df.drop(df[df['A'].str[-2:].isin(['06', '09'])].index, inplace=True)
print(df)
```
输出结果为:
```
A B
2 ghi07 3
3 jkl10 4
```
解释一下代码:
1. 首先创建一个示例数据,其中列A的取值分别是"abc06"、"def09"、"ghi07"、"jkl10"。
2. 使用`str`属性提取列A取值的后两个字符,然后使用`isin`方法判断是否等于"06"或"09",返回一个布尔型的Series。
3. 使用这个布尔型Series作为索引,找到符合条件的行的位置,然后使用`drop`方法删除这些行。
4. 最后输出删除后的结果。
相关问题
如果dataframe列A取值为NaN,那么让生成列B=1;dataframe列A取值不是NaN,就生成列B=列A+1
您好,欢迎使用ChitGPT进行对话。回答您的问题,您可以使用pandas库中的fillna()方法来实现。代码如下:
```
import pandas as pd
df['B'] = df['A'].fillna(1)
df.loc[df['A'].notnull(), 'B'] = df['A']
```
在第一行中,我们使用fillna()方法将列A中的NaN值替换为1并将结果存储在列B中。在第二行中,我们使用notnull()方法来检测A列是否包含非NaN值,如果是,则将A列的值赋给B列。
请注意,以上代码仅适用于DataFrame类型的数据。如果您有任何其他问题,请告诉我。
如果dataframe列A取值为空,那么让生成列B=列A+1;否则就生成列B=列A
你可以使用如下代码来实现:
```python
import pandas as pd
# 创建 dataframe
df = pd.DataFrame({
'A': ['a', '', 'b', 'c', ''],
})
# 生成列B
df['B'] = df['A'].apply(lambda x: x+'1' if x == '' else x)
# 打印 dataframe
print(df)
```
输出结果为:
```
A B
0 a a
1 1
2 b b
3 c c
4 1
```
其中,使用 `apply` 方法将一个函数应用到每个元素上,根据列A是否为空来生成列B的值。