pandas 加载一个表格 并进行封装
时间: 2023-07-25 22:43:33 浏览: 50
好的,可以使用 Pandas 库中的 read_csv() 方法来加载一个表格,然后进行封装。例如,假设我们有一个名为 data.csv 的 csv 文件,它包含以下数据:
```
name,age,city
John,25,New York
Emily,30,San Francisco
Tom,35,Los Angeles
```
我们可以使用以下代码加载并封装这个表格:
```python
import pandas as pd
class Table:
def __init__(self, filepath):
self.df = pd.read_csv(filepath)
def get_row(self, index):
return self.df.loc[index]
def get_column(self, col_name):
return self.df[col_name]
def get_cell(self, row_index, col_name):
return self.df.loc[row_index, col_name]
```
这个 Table 类中,我们在初始化时使用 read_csv() 方法加载表格,并将其存储在一个 DataFrame 对象中。然后,我们实现了三个方法来获取一行、一列或一个单元格的数据。例如,我们可以使用以下代码来使用这个 Table 类:
```python
table = Table('data.csv')
print(table.get_row(1)) # 获取第二行数据
print(table.get_column('age')) # 获取 age 列数据
print(table.get_cell(0, 'city')) # 获取第一行 city 列数据
```
输出结果如下:
```
name Emily
age 30
city San Francisco
Name: 1, dtype: object
0 25
1 30
2 35
Name: age, dtype: int64
New York
```