クラスター分析

############################
# クラスター分析
#      coded by Y.Suganuma
############################

#############################################
# 2つのデータ間の距離
#      method : =0 : ユークリッド距離
#               =1 : ユークリッド距離の2乗
#      i,j : データ番号
#      n : 変量の数
#      x : データ
#      return : 距離
#############################################
def rng(method, i, j, n, x)

	r = 0.0

	for i2 in 0 ... n
		x1  = x[i][i2] - x[j][i2]
		r  += x1 * x1
	end
	if method == 0
		r = Math.sqrt(r)
	end

	return r
end

###############################################
# 2つのクラスター間の距離
#      method : =1 : 最短距離法
#               =2 : 最長距離法
#               =3 : メジアン法
#               =4 : 重心法
#               =5 : 群平均法
#               =6 : ウォード法
#      dio : クラスターiとクラスターoとの距離
#      djo : クラスターjとクラスターoとの距離
#      dij : クラスターiとクラスターjとの距離
#      ni : クラスターiに含まれるデータ数
#      nj : クラスターjに含まれるデータ数
#      no : クラスターoに含まれるデータ数
#      x,y : データ
#      return : 距離
#############################################
def rng_c(method, dio, djo, dij, ni, nj, no)

	r = 0.0
			# 最短距離法
	if method == 1
		if dio <= djo
			r = dio
		else
			r = djo
		end
			# 最長距離法
	elsif metho == 2
		if dio >= djo
			r = dio
		else
			r = djo
		end
			# メジアン法
	elsif metho == 3
		r = 0.5 * dio + 0.5 * djo - 0.25 * dij
			# 重心法
	elsif metho == 4
		nk = ni + nj
		r  = ni * dio / nk + nj * djo / nk - Float(ni * nj * dij) / Float(nk * nk)
			# 群平均法
	elsif metho == 5
		nk = ni + nj
		r  = ni * dio / nk + nj * djo / nk
			# ウォード法
	elsif metho == 6
		nk = ni + nj
		r  = ((ni + no) * dio + (nj + no) * djo - no * dij) / Float(nk + no)
	end

	return r
end

#################################
# クラスター分析
#      method : =1 : 最短距離法
#               =2 : 最長距離法
#               =3 : メジアン法
#               =4 : 重心法
#               =5 : 群平均法
#               =6 : ウォード法
#      ll : グループの数
#      nn : データの数
#      n : 変量の数
#      x : データ
#      g : 所属するグループ番号
#      coded by Y.Suganuma
#################################

def cluster(method, ll, nn, n, x, g)

			# 初期設定
	ci = 0
	cj = 0
	mm = nn
	if method < 4
		k = 0
	else
		k = 1
	end
	n_g = Array.new(nn)
	for i1 in 0 ... nn
		n_g[i1] = 1
	end
	r = Array.new(nn)
	for i1 in 0 ... nn
		r[i1] = Array.new(nn)
		g[i1] = i1
		for i2 in i1+1 ... nn
			r[i1][i2] = rng(k, i1, i2, n, x)
		end
	end
	for i1 in 0 ... nn
		for i2 in i1+1 ... nn
			r[i2][i1] = r[i1][i2]
		end
	end
			# 実行
	while mm > ll
				# 最小距離のクラスターを探す
		min = -1.0
		for i1 in 0 ... nn
			if g[i1] == i1
				for i2 in i1+1 ... nn
					if g[i2] == i2
						if min < 0.0 or r[i1][i2] < min
							min = r[i1][i2]
							ci  = i1
							cj  = i2
						end
					end
				end
			end
		end
				# クラスターを融合し,他のクラスターとの距離を計算
		for i1 in 0 ... nn
			if g[i1] == i1
				for i2 in i1+1 ... nn
					if g[i2] == i2
						if i1 != cj and i2 != cj
							if i1 == ci
								r[i1][i2] = rng_c(method, r[ci][i2], r[cj][i2], r[ci][cj], n_g[ci], n_g[cj], n_g[i2])
								r[i2][i1] = r[i1][i2]
							elsif i2 == ci
								r[i1][i2] = rng_c(method, r[i1][ci], r[i1][cj], r[ci][cj], n_g[ci], n_g[cj], n_g[i2])
								r[i2][i1] = r[i1][i2]
							end
						end
					end
				end
			end
		end

		n_g[ci] += n_g[cj]
		for i1 in 0 ... nn
			if g[i1] == cj
				g[i1] = ci
			end
		end

		mm -= 1
	end
end

ss     = gets().split(" ")
method = Integer(ss[0])   # 方法
ll     = Integer(ss[1])   # グループ数
n      = Integer(ss[2])   # 変数の数
nn     = Integer(ss[3])   # データの数

g      = Array.new(nn)
x      = Array.new(nn)
for i1 in 0 ... nn
	x[i1] = Array.new(n)
end

for i1 in 0 ... nn   # データ
	ss = gets().split(" ")
	for i2 in 0 ... n
		x[i1][i2] = Float(ss[i2])
	end
end

cluster(method, ll, nn, n, x, g)

k = 1;
while k <= ll
	k1 = -1
	print("グループ " + String(k) + "\n");
	for i1 in 0 ... nn
		if g[i1] >= 0
			k1 = g[i1]
			break
		end
	end
	for i1 in 0 ... nn
		if g[i1] == k1
			print("   " + String(i1))
			for i2 in 0 ... n
				print(" " + String(x[i1][i2]))
			end
			print("\n")
			g[i1] = -1
		end
	end
	k += 1
end

=begin
---------データ例(コメント部分を除いて下さい)---------
1 3 2 100   # クラスター間距離を計算する方法(method),クラスタの数(ll,クラスタの数がこの値になったら終了),変数の数(n),及び,データの数(nn)
66.813742 25.509139   # x, y
50.813965 25.537399
74.048072 39.627578
14.792737 67.086620
42.838304 11.391394
32.301288 68.001189
24.436877 42.170217
60.451432 36.760112
24.314984 30.052427
55.573736 63.561140
72.162565 19.845151
43.217518 90.320780
51.888313 53.074269
28.150842 40.674284
31.619668 57.984410
64.360526 49.975656
53.702823 79.538952
45.052136 35.297650
4.702717 49.420025
92.649989 5.431771
65.403044 56.028507
48.797150 30.888271
40.806244 14.484027
84.355436 53.896156
50.133508 76.624964
49.405414 78.403179
76.824779 41.008448
63.246400 40.075843
78.538747 62.910356
64.396460 56.299403
77.152853 44.325319
80.592411 36.177862
48.763290 43.966714
58.789819 90.533976
13.684999 67.266671
75.163001 81.909246
46.451671 56.376068
65.105023 34.563928
45.815747 20.164535
33.227702 44.477025
41.061314 -54.334662
40.021202 -65.490282
22.829721 -56.861792
43.917033 -43.558827
40.723550 -43.188580
53.309317 -69.996428
31.463381 -31.506555
24.313168 -52.662580
17.433859 -57.468181
32.457761 -93.127719
9.183688 -31.686863
10.186332 -44.173216
27.925638 -62.236940
9.356182 -42.291381
23.036540 -27.153692
-16.261328 -35.185462
29.557288 -31.933335
36.519228 -59.269642
25.692837 -57.033021
22.797337 -52.169044
15.312205 -70.370841
22.086548 -25.219727
50.448475 -34.825824
30.814475 -40.968689
28.798789 -65.737936
59.822791 -25.673863
16.224354 -48.408079
6.867234 -61.804123
44.250439 -50.598095
54.117605 -20.539278
-43.244836 -3.374350
-19.122981 -9.026765
-4.632598 27.480272
26.647517 24.535738
19.527239 1.573533
-29.982384 26.884977
-11.856633 -24.568687
-9.244616 -3.343794
-15.656771 32.298241
-22.152364 24.203428
-34.526986 -27.862305
-60.500060 -18.400603
-6.902336 21.631805
-46.863390 -27.364383
-36.999128 4.181449
-24.599285 -17.177658
-24.814923 9.906257
-7.845930 9.388576
-36.128527 5.644606
-31.522759 11.282073
-12.500001 24.953627
-21.330455 -0.830663
-21.270652 -1.890803
-23.750951 5.435622
-25.563039 22.105947
-16.047432 -5.815069
-34.280972 30.657254
-36.264330 -8.759981
-12.109507 -15.804752
-33.227330 -5.857611
=end