Python Pandas 列数

Python Pandas 列数

在数据分析和处理过程中，Pandas 是一个非常强大的 Python 库，它提供了数据结构和数据分析工具，其中最重要的数据结构是 DataFrame。DataFrame 是一个二维标记数组，它由行和列组成，每列可以包含不同的数据类型，为了更好地理解 Pandas 中的列数操作，我们将通过实例来详细介绍。

创建一个简单的 DataFrame

首先，我们来创建一个简单的 DataFrame，以便后续进行列数操作。

import pandas as pd

data = {
    'Name': ['Alice', 'Bob', 'Charlie', 'David'],
    'Age': [25, 30, 35, 40],
    'City': ['New York', 'Los Angeles', 'Chicago', 'Houston']
}

df = pd.DataFrame(data)
print(df)

运行结果：

      Name  Age         City
0    Alice   25     New York
1      Bob   30  Los Angeles
2  Charlie   35      Chicago
3    David   40      Houston

获取 DataFrame 的列数

要获取 DataFrame 的列数，可以使用 shape 属性，并取其第二个元素。

num_columns = df.shape[1]
print("Number of columns in DataFrame: ", num_columns)

运行结果：

Number of columns in DataFrame:  3

添加新的列

在 Pandas 中，可以通过给 DataFrame 直接赋值的方式来添加新的列。

df['Country'] = ['USA', 'USA', 'USA', 'USA']
print(df)

运行结果：

      Name  Age         City Country
0    Alice   25     New York     USA
1      Bob   30  Los Angeles     USA
2  Charlie   35      Chicago     USA
3    David   40      Houston     USA

删除列

要删除 DataFrame 中的列，可以使用 drop 方法，并指定要删除的列名和 axis 参数。

df.drop('Country', axis=1, inplace=True)
print(df)

运行结果：

      Name  Age         City
0    Alice   25     New York
1      Bob   30  Los Angeles
2  Charlie   35      Chicago
3    David   40      Houston

选择特定的列

要选择 DataFrame 中的特定列，可以直接使用列名作为索引。

selected_columns = df[['Name', 'City']]
print(selected_columns)

运行结果：

      Name         City
0    Alice     New York
1      Bob  Los Angeles
2  Charlie      Chicago
3    David      Houston

以上就是关于 Python Pandas 列数的介绍，通过这些操作，我们可以轻松地对 DataFrame 中的列进行增加、删除、选择等操作。