python从dataframe中取出列中特定行的值为1的列
时间: 2024-05-11 11:15:44 浏览: 13
可以使用 Pandas 库的 loc 方法和布尔索引来实现。假设要取出 DataFrame 中名为 column_name 的列中,值为 1 的所有行的数据,可使用以下代码:
```
import pandas as pd
# 创建一个 DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [0, 1, 1], 'C': [1, 0, 1]})
# 使用布尔索引筛选出符合条件的所有行
result = df.loc[df[column_name] == 1]
```
其中,df[column_name] 取出名为 column_name 的列,返回一个 Series 对象。该对象与一个值比较会返回一个布尔类型的 Series,其中符合条件的位置值为 True,不符合的为 False。使用该布尔数组来筛选出符合条件的行,即可得到一个新的 DataFrame,其中只有符合条件的行和全部列。
相关问题
python从dataframe中取出特定几行
这是一个关于编程的问题,可以回答。要从 DataFrame 中取出特定的几行,可以使用 loc 或 iloc 函数。loc 函数根据行标签和列标签进行索引,iloc 函数根据行索引和列索引进行索引。例如,要取出第 0 行到第 2 行,可以使用以下代码:
```
df.loc[0:2, :]
```
或者
```
df.iloc[0:3, :]
```
其中 ":" 表示选取所有列。
python从dataframe中取出特定几列
对于您的问题,我可以提供以下代码示例:
假设您的数据框名为df,要选择的列为column1、column2、column3,可以使用以下代码:
```
new_df = df[['column1', 'column2', 'column3']]
```
这将创建一个新的数据框new_df,其中仅包含您要选择的列。