题目来源于力扣:
2878. 获取 DataFrame 的大小 - 力扣(LeetCode)
题目要求:
DataFrame players:
+-------------+--------+
| Column Name | Type |
+-------------+--------+
| player_id | int |
| name | object |
| age | int |
| position | object |
| ... | ... |
+-------------+--------+
编写一个解决方案,计算并显示 players 的 行数和列数。
将结果返回为一个数组:
[number of rows, number of columns]
思路流程:
- 获取一个数组的大小,可以直接想到使用.shape函数进行求解,使用形式如下
#data是一个三行五列的数据框
print list(data.shape)
#生成 [3,5] 的列表
- 或者我们可以只用len函数,len函数可以显示出数据框的大小,如果直接把数据框的名字传入len函数中,系统会自动默认生成行的数量。如果想返回列的数量,只需要在传入的数据框名字后添加“.columns” 即可,表达按照列的数量进行计算,使用形式如下
#data是一个三行五列的数据框
a=(len(data), len(data.columns))
print list(a)
#生成 [3,5] 的列表
- 我们也可以通过len搭配count函数进行行列数的求解。count可以求得每行/列不为空值的数量,生成一个序列,而len函数可以计算这些序列的长度,从而完成目的
#data是一个三行五列的数据框
a=(len(data.count(axis=1)), len(data.count(axis=0)))
print list(a)
#生成 (3,5) 的元组
注意,count的参数axis中,1行0列