用python将dataframe中的特定两列放到新的dataframe中
时间: 2023-03-30 22:01:09 浏览: 119
好的,这个问题很简单。您可以使用 Pandas 库中的 DataFrame 类的 loc 方法来实现。具体代码如下:
```python
import pandas as pd
# 假设您的原始 DataFrame 名称为 df,需要提取的两列分别为 column1 和 column2
new_df = df.loc[:, ['column1', 'column2']]
```
这样就可以将特定两列放到新的 DataFrame 中了。希望能对您有所帮助。
相关问题
将dataframe中的两列画的折线图放到同一图中
好的,可以使用matplotlib库来实现这个需求。首先,需要在代码中导入matplotlib库和pandas库。
下面是一个简单的示例代码,假设我们有一个名为df的DataFrame,其中有两列分别为'A'和'B':
```python
import matplotlib.pyplot as plt
import pandas as pd
# 创建一个DataFrame
df = pd.DataFrame({'A': [1, 2, 3, 4, 5], 'B': [2, 4, 6, 8, 10]})
# 创建一个画布
fig, ax = plt.subplots()
# 在同一张图中绘制'A'和'B'列的折线图
ax.plot(df['A'], label='A')
ax.plot(df['B'], label='B')
# 添加图例
ax.legend()
# 展示图像
plt.show()
```
在这个示例代码中,我们先创建了一个DataFrame,并且在同一张图中绘制了'A'和'B'列的折线图。最后,添加了图例并展示了图像。你可以根据自己的数据和需求进行调整。
python如何提取dataframe中的两列作为一个列表
可以使用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]]
```