将两列数据写入DataFrame中
时间: 2024-03-24 16:38:23 浏览: 92
可以使用Python中的pandas库来创建DataFrame,并将两列数据写入其中。
首先需要安装pandas库,可以在终端中输入以下命令进行安装:
```
pip install pandas
```
安装完成后,在Jupyter Notebook中输入以下代码来创建DataFrame并将两列数据写入其中:
```python
import pandas as pd
# 定义数据
data = {'姓名': ['小明', '小红', '小刚', '小李'],
'年龄': [20, 18, 22, 19],
'性别': ['男', '女', '男', '男']}
# 创建DataFrame对象
df = pd.DataFrame(data)
# 将“姓名”和“年龄”这两列数据写入新的DataFrame中
new_df = df[['姓名', '年龄']]
# 显示新的DataFrame
new_df
```
运行代码后,就可以在Jupyter Notebook中显示出只包含“姓名”和“年龄”这两列数据的DataFrame,如下图所示:
![pandas两列数据](https://img-blog.csdnimg.cn/20210912113303957.png)
相关问题
将列表中的两列数据写入新的dataframe中
可以使用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)
python将两列数据写入csv
在Python中,你可以使用内置的`csv`模块或者第三方库如`pandas`来将两列数据写入CSV文件。这里我们分别介绍这两种方法。
**使用csv模块**:
```python
import csv
# 假设data是一个包含两列数据的列表,例如:
data = [['名字', '年龄'], ['张三', '25'], ['李四', '30']]
with open('output.csv', 'w', newline='') as csvfile:
writer = csv.writer(csvfile)
# 写入表头
writer.writerow(data[0])
# 写入数据
for row in data[1:]:
writer.writerow(row)
```
**使用pandas库**:
```python
import pandas as pd
# 创建DataFrame
df = pd.DataFrame({
'名字': ['张三', '李四'],
'年龄': [25, 30]
})
# 将DataFrame保存为CSV文件
df.to_csv('output.csv', index=False)
```
在这两个例子中,`output.csv`将是写入的数据,每行表示一列,第一行通常表示列名。
阅读全文