python如何提取dataframe中的两列作为一个列表
时间: 2023-03-30 10:00:38 浏览: 72
可以使用pandas库中的iloc方法来提取dataframe中的两列作为一个列表。具体代码如下:
```python
import pandas as pd
# 创建一个示例dataframe
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})
# 提取'A'和'B'列作为一个列表
lst = df.iloc[:, [, 1]].values.tolist()
print(lst)
```
输出结果为:
```
[[1, 4], [2, 5], [3, 6]]
```
相关问题
python中取DataFrame中两列画柱状图
好的,可以使用pandas和matplotlib库来实现。首先,假设你的DataFrame变量名为df,要绘制的两列分别为col1和col2,可以按照以下步骤进行操作:
1. 导入相关库
```python
import pandas as pd
import matplotlib.pyplot as plt
```
2. 选取两列数据
```python
data = df[['col1', 'col2']]
```
3. 绘制柱状图
```python
data.plot(kind='bar')
plt.show()
```
完整的代码如下:
```python
import pandas as pd
import matplotlib.pyplot as plt
data = df[['col1', 'col2']]
data.plot(kind='bar')
plt.show()
```
这样就可以绘制出DataFrame中col1和col2两列的柱状图了。
python dataframe中两列数据相加得到一列新的数据
你可以使用 pandas 库中的 DataFrame 的 add() 方法将两列数据相加,然后将结果存储在一个新的列中。例如:
```python
import pandas as pd
# 创建一个 DataFrame
df = pd.DataFrame({'A':[1,2,3],'B':[4,5,6]})
# 将 A 列和 B 列相加,存储在 C 列中
df['C'] = df['A'].add(df['B'])
# 打印结果
print(df)
```
输出结果为:
```
A B C
0 1 4 5
1 2 5 7
2 3 6 9
```
在这个例子中,将 DataFrame 的 A 列和 B 列相加,并将结果存储在一个新的列 C 中。