Question

我有一张表格如下：

  1 2 3 4 5 6 . . . .
1 1 0 0 0 1 0 . . . .
2 0 0 1 1 1 1 . . . .
3 0 1 0 0 0 1 . . . .
4 1 0 0 0 0 0 . . . .
5 0 0 1 0 1 0 . . . .
. . . . . . . . . . .
. . .
. .
.

1,2，....是行和列的标题。我需要索引到表中这意味着：一个数组（向量），索引到row1，包含1（=列1，因为在表格单元格（1,1）中为真）。另一个索引到row2的数组包含3,4,5,6（因为单元格（2,3），（2,4），（2,5），（2,6）是真的）等等...

我准确地阅读了framework-bundle和Compact MATLAB matrix indexing notation，但我无法为其编写代码。

Answer 1

由于每个结果数组的大小不同，因此可以使用单元格数组首先，你的样本数据实际上不是一个表，所以让我们做一个任意的：

T = table({'a' 'a' 'a' 'b' 'b'}.',{'X' 'Y' 'Z' 'X' 'Z'}.',(1:5).',...
    'VariableNames',{'UserId','productId','Rating'});

接下来，我们将所有'key'列转换为分类数组：

T.UserId = categorical(T.UserId);
T.productId = categorical(T.productId);

然后我们使用这个分类数组来交叉表格：

cross_T = crosstab(T.UserId,T.productId)

现在我们在新矩阵中查找所有1：

[r,c] = find(cross_T);

并使用arrayfun按行收集它们：

% a function to return all 1 in row
row = @(x) c(r==x).';
% preform on all rows
r = arrayfun(row,1:size(cross_T,1),'UniformOutput',false).';

因此我们将单元格数组r作为输出：

r = 
    [1x3 double]
    [1x2 double]

要查找我们写的特定用户的数据：

>> r{2}
ans =
     1     3

您是否希望将其转换为结构数组更具可读性：

s = cell2struct(r,categories(T.UserId))

那么s的输出将是：

s = 
    a: [1 2 3]
    b: [1 3]

Answer 2

假设你有以下矩阵

>> A = randi([0,1], [5,5])
A =

   1   0   1   1   1
   1   0   1   0   1
   1   1   1   1   0
   0   1   1   0   1
   0   0   0   1   0

您可以通过

分别找到每行的向量

>> find(A(1,:))
ans =

   1   3   4   5

如果你想收集这些向量，你需要决定你想要收集它们的结构类型。

将索引创建为双表以便快速访问

2 个答案: