在Python中,pandas
库是一个非常流行的数据处理库,它提供了DataFrame
这一数据结构来高效地处理表格化的数据。如果想查看一个DataFrame的行数和列数,可以使用.shape来实现。
一、基本用法
当你有一个名为df
的DataFrame时,只需调用.shape
属性,它将返回一个包含两元素的元组,其中第一个元素是DataFrame的行数,第二个元素是列数。
import pandas as pd
# 假设df是你的DataFrame
df = pd.DataFrame({
'Column1': [1, 2, 3],
'Column2': ['A', 'B', 'C'],
'Column3': [True, False, True]
})
# 使用shape属性查看行数和列数
row_count, column_count = df.shape
print("行数:", row_count)
print("列数:", column_count)
输出为:
行数: 3
列数: 3
注意事项
.shape
属性是直接从NumPy数组继承而来的一个特性,在很多其他情境下(如处理NumPy数组时)也会用到它来获取数组的维度信息。- 虽然我们主要关注的是行数和列数,但
.shape
提供的信息本质上是多维数据结构的维度信息,这对于理解数据的形状非常有用。
二、每日一题
编写一个解决方案,计算并显示 players
的 行数和列数。
将结果返回为一个数组:
[number of rows, number of columns]
import pandas as pd
def getDataframeSize(players: pd.DataFrame) -> List[int]:
p_shape = [players.shape[0], players.shape[1]]
return p_shape
定义了一个名为 getDataframeSize 的新函数,该函数将 DataFrame players 作为输入参数,并返回一个包含 DataFrame players 中的行数和列数的列表。
题目来源:力扣(LeetCode)