pd.DataFrame用法
时间: 2023-11-03 17:59:02 浏览: 59
`pd.DataFrame` 是 pandas 库中的一个类,用于创建和操作二维表格数据。下面是一些常用的用法:
### 创建 DataFrame
可以从各种数据源创建 DataFrame,例如:
- 从 list 或 ndarray 创建 DataFrame:
```python
import pandas as pd
data = [['Alice', 20], ['Bob', 25], ['Charlie', 30]]
df = pd.DataFrame(data, columns=['Name', 'Age'])
print(df)
```
输出:
```
Name Age
0 Alice 20
1 Bob 25
2 Charlie 30
```
- 从 dictionary 创建 DataFrame:
```python
import pandas as pd
data = {'Name': ['Alice', 'Bob', 'Charlie'], 'Age': [20, 25, 30]}
df = pd.DataFrame(data)
print(df)
```
输出:
```
Name Age
0 Alice 20
1 Bob 25
2 Charlie 30
```
### 数据操作
DataFrame 支持各种数据操作,例如:
- 选择行或列:
```python
import pandas as pd
data = {'Name': ['Alice', 'Bob', 'Charlie'], 'Age': [20, 25, 30]}
df = pd.DataFrame(data)
# 选择第一行
print(df.loc[0])
# 选择 'Name' 列
print(df['Name'])
```
输出:
```
Name Alice
Age 20
Name: 0, dtype: object
0 Alice
1 Bob
2 Charlie
Name: Name, dtype: object
```
- 添加新列:
```python
import pandas as pd
data = {'Name': ['Alice', 'Bob', 'Charlie'], 'Age': [20, 25, 30]}
df = pd.DataFrame(data)
# 添加新列 'Gender'
df['Gender'] = ['F', 'M', 'M']
print(df)
```
输出:
```
Name Age Gender
0 Alice 20 F
1 Bob 25 M
2 Charlie 30 M
```
- 删除行或列:
```python
import pandas as pd
data = {'Name': ['Alice', 'Bob', 'Charlie'], 'Age': [20, 25, 30]}
df = pd.DataFrame(data)
# 删除第一行
df = df.drop(0)
# 删除 'Age' 列
df = df.drop('Age', axis=1)
print(df)
```
输出:
```
Name
1 Bob
2 Charlie
```
### 数据统计
DataFrame 支持各种数据统计,例如:
- 求和:
```python
import pandas as pd
data = {'Name': ['Alice', 'Bob', 'Charlie'], 'Age': [20, 25, 30]}
df = pd.DataFrame(data)
# 计算 'Age' 列的和
print(df['Age'].sum())
```
输出:
```
75
```
- 求均值:
```python
import pandas as pd
data = {'Name': ['Alice', 'Bob', 'Charlie'], 'Age': [20, 25, 30]}
df = pd.DataFrame(data)
# 计算 'Age' 列的均值
print(df['Age'].mean())
```
输出:
```
25.0
```
- 求标准差:
```python
import pandas as pd
data = {'Name': ['Alice', 'Bob', 'Charlie'], 'Age': [20, 25, 30]}
df = pd.DataFrame(data)
# 计算 'Age' 列的标准差
print(df['Age'].std())
```
输出:
```
4.08248290463863
```
这只是一些常用的用法,pandas 还有很多功能和用法,可以参考官方文档了解更多。
阅读全文