使用dataframe列上的条件填充数组

Question

我想要计算一列中的行的频率，如果频率大于3，我想用这些行填充一个数组。

例如：

402       Sony
403       Sony
404       Sony
405        ZTE
406        ZTE
407        ZTE
408        ZTE
409        ZTE

所需的数组：

[
"ZTE",
"ZTE",
"ZTE",
"ZTE",
"ZTE"]

Answer 1

使用collections.Counter()获取出现频率。因此，假设您提供的数据来自文件test.txt

import collections
itemList = []
with open("test.txt", 'r') as f:
    for line in f:
        itemList.append(line.split()[1])
result = collections.Counter(itemList)
print(result)
#get the n most repeated values in a list of tuples:
n = 1
print(result.most_common(n))

将输入

402       Sony
403       Sony
404       Sony
405        ZTE
406        ZTE
407        ZTE
408        ZTE
409        ZTE

和输出：

Counter({'ZTE': 5, 'Sony': 3})
[('ZTE', 5)]