将列表中的两列数据写入新的dataframe中
时间: 2023-10-02 13:06:11 浏览: 99
Python使用Excel将数据写入多个sheet
可以使用Python中的pandas库来创建DataFrame,并将列表中的两列数据写入新的DataFrame中。
首先需要安装pandas库,可以在终端中输入以下命令进行安装:
```
pip install pandas
```
安装完成后,在Jupyter Notebook中输入以下代码来创建DataFrame并将列表中的两列数据写入新的DataFrame中:
```python
import pandas as pd
# 定义数据列表
data = [['小明', 20, '男'], ['小红', 18, '女'], ['小刚', 22, '男'], ['小李', 19, '男']]
# 创建DataFrame对象
df = pd.DataFrame(data, columns=['姓名', '年龄', '性别'])
# 将“姓名”和“年龄”这两列数据写入新的DataFrame中
new_df = df[['姓名', '年龄']]
# 显示新的DataFrame
new_df
```
运行代码后,就可以在Jupyter Notebook中显示出只包含“姓名”和“年龄”这两列数据的DataFrame,如下图所示:
![pandas两列数据列表](https://img-blog.csdnimg.cn/20210912114054820.png)
阅读全文