2017-01-30 5 views
2

Von R Dokumentation:Sweep Äquivalent in Julia

sweep: ein Array von einer Eingangsanordnung erhalten Return durch eine Zusammenfassung Statistik Fegen.

Zum Beispiel, hier ist, wie ich jede Zeile durch die Zeilensumme teilen:

> rs = rowSums(attitude) 
> ratios = sweep(attitude, 1, rs, FUN="/") 
> head(ratios) 
    rating complaints privileges learning raises critical advance 
1 0.1191136 0.1412742 0.08310249 0.1080332 0.1689751 0.2548476 0.12465374 
2 0.1518072 0.1542169 0.12289157 0.1301205 0.1518072 0.1759036 0.11325301 
3 0.1454918 0.1434426 0.13934426 0.1413934 0.1557377 0.1762295 0.09836066 
4 0.1568123 0.1619537 0.11568123 0.1208226 0.1388175 0.2159383 0.08997429 
5 0.1680498 0.1618257 0.11618257 0.1369295 0.1473029 0.1721992 0.09751037 
6 0.1310976 0.1676829 0.14939024 0.1341463 0.1646341 0.1493902 0.10365854 
> rowSums(ratios) # check that ratios sum up to 1 
[1] 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 

Mein Versuch in Julia:

x = rand(3, 4) 
x[1, 1] = 10 
x[2, 1] = 20 
x[3, 1] = 30 
rowsum = sum(x, 2) 
rowsum_mat = repmat(rowsum, 1, size(x, 2)) 
x = x ./ rowsum_mat 

Das funktioniert aber ist klobig. Gibt es einen eleganteren und effizienteren Weg dies zu tun?

+0

R-Tag entfernen. Der Sweep-Operator ist nicht spezifisch für R. –

Antwort

8

Keine Notwendigkeit zu verwenden repmat - alle von Julia . -Operatoren tun "broadcasting" standardmäßig. Dies bedeutet, dass die Dimensionen der beiden Argumente übereinstimmen und alle Dimensionen mit der Länge 1 (die Singleton-Dimensionen) auf das andere Array erweitert werden. Da Reduktionen die gleiche Dimensionalität der Quellmatrix beibehalten, können sie direkt mit jedem Punktoperator verwendet werden.

In Ihrem Fall können Sie nur verwenden:

x ./ sum(x, 2) 

seit:

julia> x ./ rowsum_mat == x ./ rowsum 
true