for文とif文を使って文字と数を返す関数

Question

前提・実現したいこと

for文、if文を活用して
str_1 = "bbbbaa"　に対して[["b", 4],["a", 2]]
str_1 = "dddddddkkk"に対して，[["d", 7], ["k", 3]]
が返される関数の作成をしたいです。

発生している問題・エラーメッセージ

文字列が"a"と"b"のみであれば個数と文字を出すことができるのですが、文字の形（"c"や"k"等）を変えた場合にも対応してくれる関数を作成したいです。

def f1(str_v):
    num_1 = ["a", 0]
    num_2 = ["b", 0]
    for i in str_v:
        if i == "a":
            num_1[1] += 1
        elif i == "b":
            num_2[1] += 1
    return num_1, num_2

Answer 1

collections.Counter()なんかを使うと、使用されている文字をそれぞれについてカウントしてくれます。
それを辞書として格納して、keyとvalueをリストに格納してやれば以下のようにして出来ます：

from collections import Counter

def f1(str_v):
	lst = []
	dic = dict(Counter(list(str(str_v))))
	for k, v in dic.items():
		lst.append([k, v])
	return lst

まああとは、itertools.groupbyなんてものもあるので、

from itertools import groupby

def f1(str_v):
	lst = []
	a = list(str_v)
	a.sort()
	for k, v in groupby(a):
		lst.append([k, len(list(v))])
	return lst

のようにしてもスッキリしますね。

Comments

[wakiyamaa]

ありがとうございます

Answer 2

collections.Counterを使うと、こうなります。

>>> import collections
>>> str_1 = "dddddddkkk"
>>> for char, count in collections.Counter(str_1).items():
...     print(f'{char} : {count}')
...
d : 7
k : 3

Comments

[wakiyamaa]

ありがとうございます。

Answer 3

dictionaryを使いましょう

Comments

[wakiyamaa]

ありがとうございます。